题目链接:hihocoder
一道拓扑排序的模板题
拓扑排序的步骤为:
1.定义一个队列,并把所有入度为0的节点加入队列
2.取出队首节点,然后删去从它出发的边,并令这些边到达的定点入度减1,如果某个顶点入度减为0,则将其加入队列。
3.反复进行2操作,直到队列为空。
4.如果队列为空时,入过队的节点数目恰好为N,说明拓扑排序成功,图为有向无环图;否则拓扑排序失败,图中有环。
#include<cstdio>
#include<vector>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn = 1e5+10;
//const int inf = 0x3f3f3f3f;
vector<int> G[maxn];
int inDeg[maxn];
int t,n,m;
bool topsort(){
int num=0; //记录加入拓扑序列的顶点数
queue<int> q;
for(int i=1;i<=n;i++){
if(inDeg[i] == 0){
q.push(i);
num++;
}
}
while(!q.empty()){
int tmp = q.front();
q.pop();
for(int i=0;i<G[tmp].size();i++){
int v = G[tmp][i];
inDeg[v]--;
if(inDeg[v]==0){
q.push(v);
num++;
}
}
G[tmp].clear();
}
if(num == n) return true;
else return false;
}
int main(){
scanf("%d",&t);
while(t--){
for(int i=0;i<=maxn;i++){
G[i].clear();
}
memset(inDeg,0,sizeof(inDeg));
scanf("%d%d",&n,&m);
int u,v;
for(int i=0;i<m;i++){
scanf("%d%d",&u,&v);
G[u].push_back(v);
inDeg[v]++;
}
if(topsort()) puts("Correct");
else puts("Wrong");
}
return 0;
}