Codefoces909E Coprocessor（拓扑排序）

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 #include<cstdlib>
 6 #include<cmath>
 7 #include<vector>
 8 #include<queue>
 9 #define IO ios::sync_with_stdio(false);cin.tie(0);
10 const int MOD=1e9+7;
11 typedef int ll;
12 using namespace std;
13 vector<int> vec[100010];
14 int n, m, indegree[100010], cnt=0, num=0, vis[100010];
15 int a[100010], x, y, tail=-1;
16 queue<int> q[2];
17 void topo()
18 {
19     for(int i = 0; i < n; i++){
20         if(!indegree[i]){
21             q[a[i]].push(i);
22         }
23     }
24     while(num<n){
25         if(!q[0].empty())
26         while(!q[0].empty()){
27             int t = q[0].front();
28             q[0].pop();
29             num++;
30             tail=t;
31             for(int i = 0; i < vec[t].size(); i++){
32                 indegree[vec[t][i]]--;
33                 if(!indegree[vec[t][i]]){
34                     q[a[vec[t][i]]].push(vec[t][i]);
35                 }
36             }
37         }
38         if(!q[1].empty()){
39             cnt++;
40             while(!q[1].empty()){
41                 int t = q[1].front();
42                 q[1].pop();
43                 num++;
44                 tail=t;
45                 for(int i = 0; i < vec[t].size(); i++){
46                     indegree[vec[t][i]]--;
47                     if(!indegree[vec[t][i]]){
48                         q[a[vec[t][i]]].push(vec[t][i]);
49                     }
50                 }
51             }    
52         }
53     }
54 }
55 int main()
56 {
57     IO;
58     memset(vis, 0, sizeof(vis));
59     cin >> n >> m;
60     for(int i = 0; i < n; i++){
61         cin >> a[i];
62     }
63     for(int i = 0; i < m; i++){
64         cin >> x >> y; //x<-y
65         vec[y].push_back(x);
66         indegree[x]++;
67     }
68     topo();
69     cout << cnt << endl;
70     return 0;
71 }