#include "bulb.h"
#include "bits/stdc++.h"
using namespace std;
template <class L, class R>
ostream &operator<<(ostream &os, pair<L, R> P) {
return os << "(" << P.first << "," << P.second << ")";
}
template <class T>
ostream &operator<<(ostream &os, vector<T> V) {
os << "[";
for (auto x : V) os << x << ", ";
return os << "]";
}
template <typename... Args>
void dbg(Args... args) {
((cerr << args << " "), ...);
cerr << "\n";
}
void dfs(int c, vector<int> &L, vector<int> &R, vector<int> &cur, vector<int> &change) {
if (L[c] == -1 || L[c] == -2) {
cur[c] = -L[c];
} else {
dfs(L[c], L, R, cur, change);
cur[c] = cur[L[c]];
}
if (R[c] == -1 || R[c] == -2) {
change[c] = -R[c];
} else {
dfs(R[c], L, R, cur, change);
change[c] = cur[R[c]];
}
}
int FindWinner(int T, std::vector<int> L, std::vector<int> R) {
int N = L.size();
vector<int> cur(N, 0), change(N, 0); // 1: R, 2: B
dfs(0, L, R, cur, change);
if (N == 1) {
if (cur[0] == 1)
return 1;
else
return 0;
}
// dbg(cur, change);
if (cur[0] == 2) return 0;
vector<int> idx;
int pv = 0;
while (pv >= 0) {
idx.push_back(pv);
pv = L[pv];
}
int i = 0;
for (; i < idx.size(); i++) {
if (change[idx[i]] == 2) break;
}
if (i == idx.size()) {
int cnt = 0;
for (int j = 0; j < i; j++) {
cnt++;
int pv = R[idx[j]];
if (pv < 0) return 1;
while (pv >= 0) {
cnt++;
pv = L[pv];
}
}
if (cnt < N) return 1;
for (int j = 0; j < i; j++) {
int pv = R[idx[j]];
if (pv == -1) return 1;
while (pv >= 0) {
if (change[pv] == 1) return 1;
cnt++;
pv = L[pv];
}
}
return 0;
}
for (int j = 0; j < i; j++) {
int pv = R[idx[j]];
int flag = true;
while (pv >= 0) {
if (change[pv] == 2) flag = false;
pv = L[pv];
}
if (flag) return 1;
}
int cnt = 0;
for (int j; j < i; j++) {
cnt++;
int pv = R[idx[j]];
while (pv >= 0) {
cnt++;
pv = L[pv];
}
}
pv = R[idx[i]];
while (pv >= 0) {
cnt++;
pv = L[pv];
}
if (cnt == N) {
pv = R[idx[i]];
while (pv >= 0) {
if (change[pv] == 2) return 0;
pv = L[pv];
}
return 1;
}
return 0;
}
Compilation message
bulb.cpp: In function 'int FindWinner(int, std::vector<int>, std::vector<int>)':
bulb.cpp:55:12: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (; i < idx.size(); i++) {
~~^~~~~~~~~~~~
bulb.cpp:58:9: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
if (i == idx.size()) {
~~^~~~~~~~~~~~~
bulb.cpp:93:12: warning: 'j' may be used uninitialized in this function [-Wmaybe-uninitialized]
for (int j; j < i; j++) {
^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
376 KB |
Output is correct |
2 |
Correct |
2 ms |
376 KB |
Output is correct |
3 |
Correct |
2 ms |
376 KB |
Output is correct |
4 |
Correct |
2 ms |
376 KB |
Output is correct |
5 |
Correct |
2 ms |
376 KB |
Output is correct |
6 |
Correct |
2 ms |
376 KB |
Output is correct |
7 |
Correct |
2 ms |
376 KB |
Output is correct |
8 |
Correct |
2 ms |
376 KB |
Output is correct |
9 |
Correct |
2 ms |
376 KB |
Output is correct |
10 |
Correct |
2 ms |
348 KB |
Output is correct |
11 |
Correct |
2 ms |
376 KB |
Output is correct |
12 |
Correct |
2 ms |
376 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
256 KB |
Output is correct |
2 |
Correct |
2 ms |
256 KB |
Output is correct |
3 |
Correct |
0 ms |
376 KB |
Output is correct |
4 |
Correct |
2 ms |
376 KB |
Output is correct |
5 |
Correct |
2 ms |
376 KB |
Output is correct |
6 |
Correct |
2 ms |
376 KB |
Output is correct |
7 |
Correct |
2 ms |
376 KB |
Output is correct |
8 |
Correct |
2 ms |
376 KB |
Output is correct |
9 |
Correct |
2 ms |
376 KB |
Output is correct |
10 |
Correct |
2 ms |
376 KB |
Output is correct |
11 |
Correct |
2 ms |
376 KB |
Output is correct |
12 |
Correct |
2 ms |
348 KB |
Output is correct |
13 |
Correct |
2 ms |
376 KB |
Output is correct |
14 |
Correct |
2 ms |
376 KB |
Output is correct |
15 |
Correct |
2 ms |
376 KB |
Output is correct |
16 |
Correct |
2 ms |
348 KB |
Output is correct |
17 |
Correct |
2 ms |
376 KB |
Output is correct |
18 |
Correct |
2 ms |
352 KB |
Output is correct |
19 |
Correct |
2 ms |
376 KB |
Output is correct |
20 |
Correct |
2 ms |
376 KB |
Output is correct |
21 |
Correct |
2 ms |
376 KB |
Output is correct |
22 |
Correct |
2 ms |
348 KB |
Output is correct |
23 |
Correct |
2 ms |
376 KB |
Output is correct |
24 |
Correct |
2 ms |
376 KB |
Output is correct |
25 |
Correct |
2 ms |
376 KB |
Output is correct |
26 |
Correct |
2 ms |
348 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
256 KB |
Output is correct |
2 |
Correct |
2 ms |
256 KB |
Output is correct |
3 |
Correct |
0 ms |
376 KB |
Output is correct |
4 |
Correct |
2 ms |
376 KB |
Output is correct |
5 |
Correct |
2 ms |
376 KB |
Output is correct |
6 |
Correct |
2 ms |
376 KB |
Output is correct |
7 |
Correct |
2 ms |
376 KB |
Output is correct |
8 |
Correct |
2 ms |
376 KB |
Output is correct |
9 |
Correct |
2 ms |
376 KB |
Output is correct |
10 |
Correct |
2 ms |
376 KB |
Output is correct |
11 |
Correct |
2 ms |
376 KB |
Output is correct |
12 |
Correct |
2 ms |
348 KB |
Output is correct |
13 |
Correct |
2 ms |
376 KB |
Output is correct |
14 |
Correct |
2 ms |
376 KB |
Output is correct |
15 |
Correct |
2 ms |
376 KB |
Output is correct |
16 |
Correct |
2 ms |
348 KB |
Output is correct |
17 |
Correct |
2 ms |
376 KB |
Output is correct |
18 |
Correct |
2 ms |
352 KB |
Output is correct |
19 |
Correct |
2 ms |
376 KB |
Output is correct |
20 |
Correct |
2 ms |
376 KB |
Output is correct |
21 |
Correct |
2 ms |
376 KB |
Output is correct |
22 |
Correct |
2 ms |
348 KB |
Output is correct |
23 |
Correct |
2 ms |
376 KB |
Output is correct |
24 |
Correct |
2 ms |
376 KB |
Output is correct |
25 |
Correct |
2 ms |
376 KB |
Output is correct |
26 |
Correct |
2 ms |
348 KB |
Output is correct |
27 |
Correct |
88 ms |
7332 KB |
Output is correct |
28 |
Correct |
87 ms |
7328 KB |
Output is correct |
29 |
Correct |
85 ms |
7288 KB |
Output is correct |
30 |
Correct |
102 ms |
23660 KB |
Output is correct |
31 |
Correct |
101 ms |
23660 KB |
Output is correct |
32 |
Correct |
87 ms |
7388 KB |
Output is correct |
33 |
Correct |
86 ms |
7288 KB |
Output is correct |
34 |
Correct |
88 ms |
7416 KB |
Output is correct |
35 |
Correct |
86 ms |
7404 KB |
Output is correct |
36 |
Correct |
87 ms |
7412 KB |
Output is correct |
37 |
Correct |
87 ms |
7324 KB |
Output is correct |
38 |
Correct |
87 ms |
7424 KB |
Output is correct |
39 |
Correct |
87 ms |
7424 KB |
Output is correct |
40 |
Correct |
87 ms |
7328 KB |
Output is correct |
41 |
Correct |
86 ms |
7332 KB |
Output is correct |
42 |
Correct |
86 ms |
7416 KB |
Output is correct |
43 |
Correct |
86 ms |
7288 KB |
Output is correct |
44 |
Correct |
85 ms |
7332 KB |
Output is correct |
45 |
Correct |
87 ms |
7332 KB |
Output is correct |
46 |
Correct |
87 ms |
7320 KB |
Output is correct |
47 |
Correct |
89 ms |
11124 KB |
Output is correct |
48 |
Correct |
91 ms |
11172 KB |
Output is correct |
49 |
Correct |
92 ms |
11180 KB |
Output is correct |
50 |
Correct |
90 ms |
11160 KB |
Output is correct |