#include "train.h"
#include <bits/stdc++.h>
using namespace std;
const int maxn = 5e3 + 20;
vector<int> adj[maxn];
bool charge[maxn];
bool good[maxn];
bool flag[maxn];
int own[maxn];
int n, m;
int root;
bool dfs1(int u) {
flag[u] = true;
for (auto v: adj[u]) {
if (v == root) {
return true;
}
else if (!flag[v] && dfs1(v)) {
return true;
}
}
return false;
}
bool dfs2(int u) {
if (good[u]) {
return true;
}
for (auto v: adj[u]) {
if (!flag[v] && dfs2(v)) {
return true;
}
}
return false;
}
vector<int> sub3() {
for (root = 0; root < n; root++) {
if (!charge[root]) {
continue;
}
for (int u = 0; u < n; u++) {
flag[u] = false;
}
good[root] = dfs1(root);
}
vector<int> res(n);
for (root = 0; root < n; root++) {
for (int u = 0; u < n; u++) {
flag[u] = false;
}
res[root] = dfs2(root);
}
return res;
}
vector<int> sub4() {
return vector<int>();
}
vector<int> who_wins(vector<int> a, vector<int> r, vector<int> u, vector<int> v) {
n = a.size();
for (int i = 0; i < n; i++) {
own[i] = a[i];
charge[i] = r[i];
}
m = u.size();
for (int i = 0; i < m; i++) {
adj[u[i]].push_back(v[i]);
}
bool is_sub3 = true;
bool is_sub4 = true;
for (int i = 0; i < n; i++) {
is_sub3 &= (own[i] == 1);
is_sub4 &= (own[i] == 0);
}
if (is_sub3) {
return sub3();
}
if (is_sub4) {
return sub4();
}
return vector<int>();
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
3 ms |
852 KB |
WA in grader: Wrong returned array size |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
420 KB |
WA in grader: Wrong returned array size |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
145 ms |
262144 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
4 ms |
1068 KB |
WA in grader: Wrong returned array size |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
5 ms |
1236 KB |
WA in grader: Wrong returned array size |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
3 ms |
852 KB |
WA in grader: Wrong returned array size |
2 |
Halted |
0 ms |
0 KB |
- |