#include <bits/stdc++.h>
#include "train.h"
#define ent '\n'
#define f first
#define s second
//#define int long long
using namespace std;
typedef long long ll;
const int maxn = 5e3 + 12;
const int mod = 1e9 + 7;
vector<int> g[maxn], rg[maxn];
bool won[maxn], del[maxn];
int out[maxn];
int n, m;
vector<int> who_wins(vector<int> a, vector<int> r, vector<int> U, vector<int> V){
n = a.size(), m = U.size();
vector<int> ans(n);
function<bool(int)> check = [&](int v){
for(int i=0;i<n;i++){
won[i] = out[i] = 0;
g[i].clear();
rg[i].clear();
}
for(int i=0;i<m;i++){
out[U[i]]++;
g[U[i]].push_back(V[i]);
rg[V[i]].push_back(U[i]);
}
queue<int> q;
won[v] = 1;
q.push(v);
while(q.size()){
int v = q.front();
q.pop();
for(int to:rg[v]){
if(a[to] == 1 && !won[to]){
won[to] = 1;
q.push(to);
}
if(a[to] == 0){
out[to]--;
if(!out[to]){
won[to] = 1;
q.push(to);
}
}
}
}
bool ok = 0;
for(int to:g[v]){
ok |= won[to];
}
if(ok){
del[v] = 1;
for(int i=0;i<n;i++){
if(won[i]){
ans[i] = 1;
r[i] = 1;
}
}
return 1;
}
return 0;
};
while(1){
bool ok = 0;
for(int i=0;i<n;i++){
if(r[i] == 1 && !del[i] && check(i)){
ok = 1;
break;
}
}
if(!ok) break;
}
return ans;
}
Compilation message
train.cpp: In lambda function:
train.cpp:24:29: warning: suggest parentheses around assignment used as truth value [-Wparentheses]
24 | won[i] = out[i] = 0;
| ~~~~~~~^~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
2075 ms |
1116 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
600 KB |
3rd lines differ - on the 2nd token, expected: '1', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
91 ms |
1632 KB |
Output is correct |
2 |
Correct |
182 ms |
1372 KB |
Output is correct |
3 |
Correct |
244 ms |
1628 KB |
Output is correct |
4 |
Execution timed out |
2068 ms |
1384 KB |
Time limit exceeded |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1769 ms |
1372 KB |
3rd lines differ - on the 1st token, expected: '1', found: '0' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
2050 ms |
1640 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
2075 ms |
1116 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |