有向图判环

有这样一道编程面试题，给一个有向图的邻接矩阵，判别它是否有环。

题目麻烦在给的邻接矩阵是以 ‘字符输入流’ 的形式给出的，所以将其处理成数字形式的是首先要做的工作。

得到邻接矩阵之后，进行拓扑排序即可。假如能完成拓扑排序那就无环，如果不能，那就是有环。

样例输入：

[[0, 1, 0], [0, 0, 1], [1, 0, 0]]

[[0, 0, 0, 1, 0], [1, 0, 0, 0, 0], [0, 0, 0, 1, 1], [0, 0, 0, 0, 0], [0, 1, 0, 0, 0]]

输出：（1代表有环，0代表无环）

1

0

#include <bits/stdc++.h> 
using namespace std; 

int main(int argc, char const *argv[]){
    string str;
    while(getline(cin, str) ){        
        vector<int> data;
        for(int i=0; i<str.size(); i++){
            if(str[i]=='0' || str[i]=='1'){
                data.push_back(str[i]-'0');
            }            
        }
                
        int n = sqrt( data.size() );                      //n是节点个数
        vector<vector<int> > graph(n,vector<int>(n,0));   //邻接矩阵 
        
        for(int i=0; i<n; i++){
            for(int j=0; j<n; j++){ 
                graph[i][j] = data[n*i+j];
            } 
        }
        
        vector<int> visited(n,0);    //n是节点个数，从上面处理后得到
        vector<int> indegree(n,0);   //各个节点的入度
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(graph[i][j]==1){
                    indegree[j]++;
                }                
            }
        } 
        queue<int> q;
        for(int i=0;i<n;i++){
            if(indegree[i]==0) 
                q.push(i);
        }
        
        int now;
        while(!q.empty()){
            now = q.front();
            q.pop();
            visited[now] = 1;
            for(int i=0;i<n;i++){
                if(!visited[i] && graph[now][i]==1){
                    indegree[i]--;
                    if(indegree[i]==0) q.push(i); 
                }
            }
        }
        
        bool sign = 0;
        for(int i=0; i<n; i++){
            if(!visited[i]){
                sign = 1;
                break;
            }            
        }
        
        cout<<sign<<endl;
    } 
        
    return 0;
}