拓扑排序模板

题目描述

判断有向无环图（DAG）以及输出拓扑序列

题目思路

• 利用queue存储入度为0的点
• 依次删去入度为0的点，再重新判断有无入度为0的点
• 若无环，应有n个点被删掉；若有环，则存在删不掉的点

代码

#include <iostream>
#include <queue>
#include <vector>
using namespace std;
#define N 501
vector<int> edge[N];//邻接表 
queue<int> Q;//入度为0的点进入队列 
int inDegree[N]; 
int main() {
    int n,m;
    while(cin>>n>>m&&n!=0){
        for(int i=0;i<n;i++){
            inDegree[i]=0;
        }
        int a,b;
        //↓输入时统计入度 
        while(m--){
            cin>>a>>b;
            inDegree[b]++;
            edge[a].push_back(b);
        }
        //↓入度为0的点放进队列 
        for(int i=0;i<n;i++){
            if(inDegree[i]==0){
                Q.push(i);
            }
        }
        //↓队列不为空，依次删掉入度为0的点，然后检测删后会不会产生新的入度为0的点 
        int cnt=0;
        while(!Q.empty()){
            int newP=Q.front();
            Q.pop();
            cnt++;
            for(int i=0;i<edge[newP].size();i++){
                inDegree[edge[newP][i]]--;
                if(inDegree[edge[newP][i]]==0){
                    Q.push(edge[newP][i]);
                }
            }
        }
        //↓如果不为n,说明存在环 
        if(cnt==n){
            puts("YES");
        }else{
            puts("NO");
        }
    }
    return 0;
}