#ifndef local
#include "train.h"
#endif
#include<bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
using namespace std;
//#define int long long
#define fi first
#define se second
#define pb push_back
#define mp make_pair
typedef pair<int, int> ii;
typedef pair<ii, int> iii;
typedef pair<ii, ii> iiii;
const int N = 3e5 + 5;
const int oo = 1e18 + 7, mod = 1e9 + 7;
int n, a[N];
mt19937 rng(1);
int rnd(int l, int r){
int temp = rng() % (r - l + 1);
return abs(temp) + l;
}
vector<int> Adj[N], Adj2[N];
bool out[N];
int deg[N], deg2[N];
vector<int> who_wins(vector<int> a, vector<int> r, vector<int> U, vector<int> V){
n = a.size();
for(int i = 0; i < U.size(); i++){
Adj[V[i]].pb(U[i]);
Adj2[U[i]].pb(V[i]);
//Adj[U[i]].pb(V[i]);
}
for(int i = 0; i < n; i++) deg2[i] = Adj2[i].size();
while(1){
vector<bool> vis(n + 1);
for(int i = 0; i < n; i++){
if(out[i]) continue;
vis[i] = r[i];
deg[i] = 0;
}
queue<int> q;
for(int i = 0; i < n; i++){
if(out[i]) continue;
if(r[i]) q.push(i);
}
while(!q.empty()){
int u = q.front();
// cout << u << "\n";
vis[u] = 1;
q.pop();
for(auto v : Adj[u]){
if(r[v]) continue;
if(out[v]) continue;
deg[v]++;
if(a[v] == 1 && deg[v] == 1) q.push(v);
if(a[v] == 0 && deg[v] == deg2[v]) q.push(v);
}
//vis[u] = 1;
//for(auto v : Adj[u]){
// if(vis[v] || out[v]) continue;
// q.push(v);
//}
}
//cout << "\n";
for(int i = 0; i < n; i++){
if(!out[i] && !vis[i]) q.push(i);
vis[i] = deg[i] = 0;
}
while(!q.empty()){
int u = q.front();
//cout << "OK " << u << "\n";
vis[u] = 1;
q.pop();
for(auto v : Adj[u]){
if(vis[v]) continue;
if(out[v]) continue;
deg[v]++;
if(a[v] == 0 && deg[v] == 1) q.push(v);
if(a[v] == 1 && deg[v] == deg2[v]) q.push(v);
}
}
//exit(0);
int cnt = 0;
for(int i = 0; i < n; i++){
if(out[i] || !vis[i]) continue;
// cout << i << "\n";
out[i] = 1;
for(auto it : Adj[i]) deg2[it]--;
cnt++;
}
if(!cnt) break;
}
vector<int> ans;
for(int i = 0; i < n; i++) ans.pb(!out[i]);
return ans;
}
//#define local
#ifdef local
void process(){
int n, m;
cin >> n >> m;
vector<int> v1, v2;
v1.resize(n), v2.resize(n);
vector<int> v3, v4;
for(int i = 0; i < n; i++) cin >> v1[i];
for(int i = 0; i < n; i++) cin >> v2[i];
for(int i = 0; i < m; i++){
int x, y;
cin >> x >> y;
v3.pb(x), v4.pb(y);
}
vector<int> v = who_wins(v1, v2, v3, v4);
for(int i = 0; i < n; i++) cout << v[i] << " ";
cout << "\n";
}
signed main(){
ios_base::sync_with_stdio(0);
cin.tie(0);
process();
}
#endif
Compilation message
train.cpp:21:21: warning: overflow in conversion from 'double' to 'int' changes value from '1.0e+18' to '2147483647' [-Woverflow]
21 | const int oo = 1e18 + 7, mod = 1e9 + 7;
| ~~~~~^~~
train.cpp: In function 'std::vector<int> who_wins(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
train.cpp:40:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
40 | for(int i = 0; i < U.size(); i++){
| ~~^~~~~~~~~~
train.cpp:79:20: warning: suggest parentheses around assignment used as truth value [-Wparentheses]
79 | vis[i] = deg[i] = 0;
| ~~~~~~~^~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
14932 KB |
Output is correct |
2 |
Correct |
11 ms |
15060 KB |
Output is correct |
3 |
Correct |
11 ms |
15056 KB |
Output is correct |
4 |
Correct |
11 ms |
15068 KB |
Output is correct |
5 |
Correct |
11 ms |
15108 KB |
Output is correct |
6 |
Correct |
13 ms |
15048 KB |
Output is correct |
7 |
Correct |
11 ms |
14984 KB |
Output is correct |
8 |
Correct |
10 ms |
15060 KB |
Output is correct |
9 |
Correct |
10 ms |
15060 KB |
Output is correct |
10 |
Correct |
11 ms |
15004 KB |
Output is correct |
11 |
Correct |
11 ms |
15060 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
7 ms |
14420 KB |
Output is correct |
2 |
Correct |
8 ms |
14420 KB |
Output is correct |
3 |
Correct |
7 ms |
14332 KB |
Output is correct |
4 |
Correct |
7 ms |
14420 KB |
Output is correct |
5 |
Correct |
7 ms |
14292 KB |
Output is correct |
6 |
Correct |
7 ms |
14420 KB |
Output is correct |
7 |
Correct |
8 ms |
14380 KB |
Output is correct |
8 |
Correct |
7 ms |
14404 KB |
Output is correct |
9 |
Correct |
7 ms |
14420 KB |
Output is correct |
10 |
Correct |
8 ms |
14408 KB |
Output is correct |
11 |
Correct |
7 ms |
14420 KB |
Output is correct |
12 |
Correct |
7 ms |
14420 KB |
Output is correct |
13 |
Correct |
8 ms |
14420 KB |
Output is correct |
14 |
Correct |
7 ms |
14420 KB |
Output is correct |
15 |
Correct |
7 ms |
14420 KB |
Output is correct |
16 |
Correct |
7 ms |
14420 KB |
Output is correct |
17 |
Correct |
7 ms |
14420 KB |
Output is correct |
18 |
Correct |
7 ms |
14432 KB |
Output is correct |
19 |
Correct |
7 ms |
14400 KB |
Output is correct |
20 |
Correct |
7 ms |
14420 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
76 ms |
15320 KB |
Output is correct |
2 |
Correct |
151 ms |
15512 KB |
Output is correct |
3 |
Correct |
203 ms |
15500 KB |
Output is correct |
4 |
Correct |
13 ms |
15444 KB |
Output is correct |
5 |
Incorrect |
13 ms |
15444 KB |
3rd lines differ - on the 854th token, expected: '1', found: '0' |
6 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
15112 KB |
Output is correct |
2 |
Correct |
12 ms |
15308 KB |
Output is correct |
3 |
Correct |
16 ms |
15444 KB |
Output is correct |
4 |
Correct |
13 ms |
15576 KB |
Output is correct |
5 |
Correct |
14 ms |
15444 KB |
Output is correct |
6 |
Correct |
13 ms |
15444 KB |
Output is correct |
7 |
Correct |
15 ms |
15444 KB |
Output is correct |
8 |
Correct |
12 ms |
15444 KB |
Output is correct |
9 |
Correct |
12 ms |
15332 KB |
Output is correct |
10 |
Correct |
13 ms |
15444 KB |
Output is correct |
11 |
Correct |
12 ms |
15448 KB |
Output is correct |
12 |
Correct |
13 ms |
15444 KB |
Output is correct |
13 |
Correct |
13 ms |
15444 KB |
Output is correct |
14 |
Correct |
12 ms |
15408 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
12 ms |
15316 KB |
3rd lines differ - on the 1st token, expected: '1', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
14932 KB |
Output is correct |
2 |
Correct |
11 ms |
15060 KB |
Output is correct |
3 |
Correct |
11 ms |
15056 KB |
Output is correct |
4 |
Correct |
11 ms |
15068 KB |
Output is correct |
5 |
Correct |
11 ms |
15108 KB |
Output is correct |
6 |
Correct |
13 ms |
15048 KB |
Output is correct |
7 |
Correct |
11 ms |
14984 KB |
Output is correct |
8 |
Correct |
10 ms |
15060 KB |
Output is correct |
9 |
Correct |
10 ms |
15060 KB |
Output is correct |
10 |
Correct |
11 ms |
15004 KB |
Output is correct |
11 |
Correct |
11 ms |
15060 KB |
Output is correct |
12 |
Correct |
7 ms |
14420 KB |
Output is correct |
13 |
Correct |
8 ms |
14420 KB |
Output is correct |
14 |
Correct |
7 ms |
14332 KB |
Output is correct |
15 |
Correct |
7 ms |
14420 KB |
Output is correct |
16 |
Correct |
7 ms |
14292 KB |
Output is correct |
17 |
Correct |
7 ms |
14420 KB |
Output is correct |
18 |
Correct |
8 ms |
14380 KB |
Output is correct |
19 |
Correct |
7 ms |
14404 KB |
Output is correct |
20 |
Correct |
7 ms |
14420 KB |
Output is correct |
21 |
Correct |
8 ms |
14408 KB |
Output is correct |
22 |
Correct |
7 ms |
14420 KB |
Output is correct |
23 |
Correct |
7 ms |
14420 KB |
Output is correct |
24 |
Correct |
8 ms |
14420 KB |
Output is correct |
25 |
Correct |
7 ms |
14420 KB |
Output is correct |
26 |
Correct |
7 ms |
14420 KB |
Output is correct |
27 |
Correct |
7 ms |
14420 KB |
Output is correct |
28 |
Correct |
7 ms |
14420 KB |
Output is correct |
29 |
Correct |
7 ms |
14432 KB |
Output is correct |
30 |
Correct |
7 ms |
14400 KB |
Output is correct |
31 |
Correct |
7 ms |
14420 KB |
Output is correct |
32 |
Correct |
76 ms |
15320 KB |
Output is correct |
33 |
Correct |
151 ms |
15512 KB |
Output is correct |
34 |
Correct |
203 ms |
15500 KB |
Output is correct |
35 |
Correct |
13 ms |
15444 KB |
Output is correct |
36 |
Incorrect |
13 ms |
15444 KB |
3rd lines differ - on the 854th token, expected: '1', found: '0' |
37 |
Halted |
0 ms |
0 KB |
- |