19011 小猿的依赖循环

Description

猿辅导2021校园招聘笔试（算法二） 
小猿在加载一个网页，这个网页共需要N个相关资源，这些资源之间有一些依赖关系。
如果这些资源中存在循环依赖，我们认为这个网页不能加载成功，否则可以加载成功。
存在循环依赖是指，这些资源中存在资源X，X依赖的资源Y直接或间接依赖于X。
你能帮助小猿判断一下这个网页能否加载成功吗？

 

输入格式

第一行输入T（T ≤ 10），表示输入T组数据。
每组数据第1行，输入一个数N（1 ≤ N ≤ 500）表示该组case有编号为1~N的N项资源。
每组数据第2到 N+1 行，输入一个 N*N 的零一矩阵。
矩阵第 i 行第 j 列数字为 a[i][j] 表示编号为 i 的资源是否依赖于编号为 j 的资源，1表示依赖，0表示不依赖。数据保证a[i][i] = 0。

 

输出格式

输出包含T行，每行输出对应每组case中是否存在循环依赖。存在输出1，不存在输出0。

思路记录：用一个sum数组记录图中节点的出度，sum[i]=a,表示节点i的出度为a,再用一个map记录节点的入度，map<a,vectory<int>>表示依赖的节点a的节点有哪些,再用一个set去维持出度为0的节点即可，写起来有点复杂，效率最大化

#include <iostream>
#include <vector>
#include<algorithm>
#include <string.h>
#include <set>
#include <map>
typedef long long ll;
using namespace std;
int main()
{
    ios::sync_with_stdio(0), cin.tie(0);
    int n, m,tem;
    cin>>n;
    while(n--){
        int sum[1005];//统计出度
        map<int,vector<int>> mp;//统计入度
        cin>>m;
        memset(sum,0,sizeof(sum));
        set<int> st;
        for(int i=1;i<=m;i++){
            for(int j=1;j<=m;j++){
                cin>>tem;
                if(tem==1){
                    sum[i]++;
                    mp[j].push_back(i);
                }
            }
        }
        for(int i=1;i<=m;i++){
            if(sum[i]==0)
                st.insert(i);
        }
        int ans=m;
        while(!st.empty()){
            ans--;
            int tem=*st.begin();
            for(int i=0;i<mp[tem].size();i++){
                sum[mp[tem][i]]--;
                if(sum[mp[tem][i]]==0) st.insert(mp[tem][i]);
            }
            st.erase(st.begin());
        }
        if(ans==0) cout<<0<<endl;
        else cout<<1<<endl;
    }
    return 0;
}