QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#615399 | #7684. Sweet Sugar | zhangboju | WA | 510ms | 138548kb | C++14 | 1.2kb | 2024-10-05 18:26:17 | 2024-10-05 18:26:17 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
constexpr int N = 2e6 + 5;
int n, k;
vector<int> g[N];
int c[N];
int f[N][2];
bool ban_to_fa[N];
int ans;
void dfs(int u, int fa) {
f[u][c[u] & 1] = c[u];
for (int v : g[u]) {
if (v == fa) continue;
dfs(v, u);
if (ban_to_fa[v]) continue;
int t0 = max(f[u][0] + f[v][0], f[u][1] + f[v][1]);
int t1 = max(f[u][0] + f[v][1], f[u][1] + f[v][0]);
f[u][0] = max(f[u][0], t0);
f[u][1] = max(f[u][1], t1);
}
if (f[u][k & 1] >= k) {
ban_to_fa[u] = 1;
++ans;
}
}
void solve() {
scanf("%d%d", &n, &k);
for (int i = 1; i <= n; i++) {
scanf("%d", &c[i]);
}
for (int i = 1; i < n; i++) {
int u, v;
scanf("%d%d", &u, &v);
g[u].push_back(v);
g[v].push_back(u);
}
dfs(1, -1);
printf("%d\n", ans);
}
void clear() {
ans = 0;
for (int i = 1; i <= n; i++) {
g[i].clear();
f[i][0] = f[i][1] = -1e9;
ban_to_fa[i] = 0;
}
}
int main() {
int T;
scanf("%d", &T);
while (T--) {
solve();
clear();
}
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 3ms
memory: 56016kb
input:
4 7 5 1 2 1 2 2 1 2 1 2 2 3 3 4 3 5 5 6 5 7 2 2 1 0 1 2 1 1 1 1 2 1
output:
2 0 1 0
result:
ok 4 number(s): "2 0 1 0"
Test #2:
score: 0
Accepted
time: 9ms
memory: 56900kb
input:
12 1 1 0 1 1 1 1 1 2 1 2 0 1 2 1 1 2 2 1 3 0 1 3 1 1 3 2 1 2000000 0 1 2000000 1 1 2000000 2
output:
0 1 0 0 0 1 0 0 0 0 0 0
result:
ok 12 numbers
Test #3:
score: 0
Accepted
time: 140ms
memory: 56664kb
input:
200000 5 2 1 1 0 0 1 2 4 5 2 4 1 3 2 5 1 0 0 0 0 0 5 1 1 2 3 2 5 4 5 3 1 0 0 0 1 1 4 4 2 3 4 5 2 5 9 1 0 0 0 2 4 3 2 1 3 1 5 1 5 3 0 1 1 0 1 5 4 2 1 4 3 5 1 5 1 0 2 1 1 1 5 3 2 4 3 4 1 4 5 1 1 0 1 1 0 1 5 4 2 1 3 5 2 5 7 0 2 1 1 2 5 1 2 3 2 5 5 4 5 5 0 1 0 1 0 2 4 4 3 5 2 1 5 5 1 0 0 1 0 1 4 1 4 5 2...
output:
1 0 0 0 1 3 3 0 0 2 0 0 2 1 0 0 1 1 0 2 0 1 0 2 1 0 0 0 0 0 1 2 0 0 2 2 0 1 0 0 0 0 3 3 0 0 1 1 2 1 2 0 4 0 1 1 0 1 0 0 1 5 0 1 1 1 0 1 1 1 1 1 1 0 1 1 1 0 3 1 0 1 0 0 4 0 0 0 1 1 0 0 1 0 2 0 5 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 1 1 0 0 0 0 1 0 0 0 0 0 1 0 0 1 1 0 1 0 1 1 1 2 0 1 2 0 0 2 0 0 1 0 0 0 0 0 ...
result:
ok 200000 numbers
Test #4:
score: 0
Accepted
time: 141ms
memory: 56700kb
input:
100000 10 16 0 1 0 1 0 1 1 0 1 1 4 10 3 10 9 1 4 7 5 2 9 5 6 9 2 3 8 4 10 18 1 0 0 1 0 0 1 1 0 2 5 1 4 2 5 3 3 4 2 10 4 9 3 8 7 3 6 5 10 13 0 1 0 1 0 1 1 0 1 1 5 2 4 3 1 6 8 9 1 8 3 5 9 4 7 6 3 10 10 16 1 0 1 0 1 1 1 0 1 1 6 8 2 4 9 6 2 6 4 10 5 3 1 6 8 7 3 6 10 13 0 0 0 0 1 0 1 0 1 0 2 1 2 7 7 9 7 ...
output:
0 0 0 0 0 0 1 0 0 1 0 0 0 0 0 1 0 1 1 0 1 1 0 0 0 1 0 0 0 0 0 0 2 1 2 1 0 2 0 0 1 0 0 2 5 1 0 2 1 0 0 1 0 0 1 1 0 1 0 1 7 0 1 1 0 0 1 1 8 1 0 1 1 1 1 0 1 1 4 0 1 0 0 0 3 0 0 2 1 10 1 1 0 2 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 0 0 0 7 0 0 2 1 0 0 0 0 0 0 0 0 0 0 0 1 0 0 5 0 6 1 2 0 1 0 0 0 0 4 1 0 0 0 7 0...
result:
ok 100000 numbers
Test #5:
score: 0
Accepted
time: 147ms
memory: 54956kb
input:
50000 20 37 2 1 0 1 1 0 0 0 1 1 1 1 1 1 0 1 1 1 0 1 3 17 14 8 8 16 18 4 14 20 7 10 10 6 2 10 13 11 5 3 6 5 6 13 11 18 1 6 12 9 9 8 15 16 19 5 9 5 20 13 0 0 0 0 1 2 1 1 0 1 0 0 0 0 0 0 1 0 0 1 17 2 3 5 11 13 12 14 13 2 19 3 1 7 12 15 6 10 17 8 20 7 1 3 3 6 13 15 16 6 3 9 9 4 13 18 3 2 20 13 0 0 0 2 0...
output:
0 0 0 0 0 0 0 0 5 1 0 1 17 3 0 0 0 1 0 1 0 0 0 1 1 0 1 0 0 0 0 1 0 1 0 1 0 1 1 0 2 3 0 1 2 1 1 0 1 1 0 0 2 0 2 0 1 1 5 1 0 1 0 15 2 3 1 0 1 2 1 1 1 2 0 1 1 1 1 1 18 1 1 2 1 2 4 0 2 1 5 0 1 0 1 1 9 0 0 0 0 1 0 8 0 0 0 1 0 3 0 0 3 1 0 0 0 1 0 0 1 0 5 0 0 0 0 0 0 0 9 0 1 0 3 0 0 0 0 0 0 3 1 1 1 1 0 2 1...
result:
ok 50000 numbers
Test #6:
score: 0
Accepted
time: 138ms
memory: 56600kb
input:
20000 50 81 1 0 1 0 0 1 0 2 1 1 0 1 1 0 0 0 1 1 1 1 1 0 1 0 1 1 1 0 1 0 1 0 1 0 1 0 0 1 0 1 0 1 1 0 1 1 0 1 0 1 45 6 43 7 29 38 3 27 42 50 27 32 50 24 41 6 12 32 17 18 6 31 44 5 2 9 2 32 28 26 2 20 40 47 34 27 16 43 4 23 36 43 9 13 19 47 4 50 25 48 37 19 49 22 42 11 46 25 1 15 10 16 8 23 6 39 30 8 2...
output:
0 3 0 0 0 0 23 0 0 0 0 0 1 0 0 1 1 0 0 0 8 0 2 4 0 6 0 0 1 0 0 1 0 1 0 0 0 0 1 0 0 1 1 1 0 0 1 0 3 1 1 1 1 3 0 0 1 2 1 1 3 1 1 1 1 0 5 1 0 0 0 1 2 1 7 4 3 2 1 0 1 0 1 0 3 1 3 1 1 1 1 1 1 0 1 1 0 0 1 0 1 0 0 0 0 0 1 0 0 0 1 0 11 0 0 1 0 0 17 3 0 4 2 0 0 2 1 2 0 0 3 1 0 0 0 0 1 0 2 0 0 2 0 0 2 1 1 1 7...
result:
ok 20000 numbers
Test #7:
score: 0
Accepted
time: 144ms
memory: 55440kb
input:
2000 500 895 1 2 0 0 1 0 1 1 0 0 0 0 1 1 0 0 1 0 1 0 1 0 1 0 0 1 0 0 0 0 1 1 1 0 0 0 0 1 1 1 0 1 1 1 0 1 0 0 1 1 0 0 1 0 0 1 1 1 1 1 0 1 1 1 0 1 1 1 1 0 0 0 0 0 1 1 0 0 0 0 0 1 1 1 1 0 0 1 1 0 0 1 0 1 0 1 1 1 0 0 1 1 0 0 1 1 0 1 0 0 0 0 0 1 1 1 1 0 0 1 0 0 1 1 2 0 0 0 1 1 0 0 0 0 0 1 1 0 1 0 1 0 0 1...
output:
0 0 6 0 0 1 0 0 1 0 1 0 35 0 0 1 0 0 0 1 0 0 1 1 4 5 0 0 0 0 0 1 0 0 0 0 3 0 0 0 0 134 0 0 3 1 12 6 9 9 1 0 0 2 2 0 1 3 0 4 11 0 1 26 0 0 3 0 2 1 0 3 5 2 3 0 1 1 12 1 1 2 36 15 1 1 0 0 2 1 1 4 1 2 0 1 0 1 1 3 0 2 0 2 0 0 0 0 0 2 1 0 0 3 3 4 0 0 0 1 0 0 0 0 0 2 0 2 1 0 0 0 0 0 0 0 0 0 2 1 4 0 1 2 9 2...
result:
ok 2000 numbers
Test #8:
score: 0
Accepted
time: 166ms
memory: 58140kb
input:
200 5000 3054 2 1 0 1 2 2 1 2 2 2 2 2 2 2 0 2 2 0 0 2 1 0 2 0 2 2 2 2 2 0 2 2 2 2 1 2 2 2 2 2 2 2 2 2 1 2 2 2 2 2 2 0 1 0 0 2 0 2 0 2 2 2 2 2 2 2 2 2 2 0 2 2 1 0 2 2 2 1 2 2 1 2 1 0 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 1 2 2 2 2 0 2 1 2 2 2 2 2 2 2 0 1 1 2 1 1 1 2 2 2 2 0 2 2 2 2 1 0 2 2 2 2 2 ...
output:
2 1 5 14 0 1 3 1 2 1 1 1 1 1 4 0 54 1 0 1 10 1 0 1 0 2 3 4 0 1 1 0 0 3 1 3 0 0 0 0 0 0 0 4 1 0 0 1 0 0 10 1 0 7 1 0 0 3 8 1 2 2 1 1 1 6 0 29 3 1 4 10 5 1 1 1 1 2 0 0 0 33 1 1 18 1 0 1 0 11 0 1 0 0 0 0 1 1 0 20 3 0 3 2 2 1 1 1 14 4 1 10 1 1 1 3 2 4 0 0 1 0 4 0 30 1 0 0 1 1 1 0 0 12 0 0 0 0 13 2 0 1 1...
result:
ok 200 numbers
Test #9:
score: 0
Accepted
time: 242ms
memory: 62400kb
input:
20 50000 24898 2 2 2 2 2 1 1 2 0 1 0 0 0 2 0 2 1 2 2 2 1 1 1 1 2 2 1 1 2 2 2 2 2 0 0 2 2 2 0 2 2 0 2 1 0 2 1 0 0 1 0 1 0 2 2 2 0 0 0 2 2 2 2 0 2 1 2 2 0 0 1 1 2 0 2 1 2 0 2 2 2 1 1 2 1 2 1 2 1 1 2 2 0 0 1 1 1 0 2 2 1 1 0 0 0 2 2 1 2 1 2 1 1 2 1 1 1 2 1 2 2 0 1 2 1 1 2 0 1 2 0 0 1 1 2 1 0 2 2 2 0 2 0...
output:
2 0 5 100 13 19 0 1 0 1 0 1 3 1 0 0 1 8 0 1
result:
ok 20 numbers
Test #10:
score: -100
Wrong Answer
time: 510ms
memory: 138548kb
input:
1 1000000 3 1 1 2 0 2 1 2 2 1 0 0 1 0 2 2 0 2 2 2 2 2 2 2 2 2 2 2 2 2 0 2 2 2 2 0 1 2 1 2 2 1 2 0 2 2 0 1 2 0 2 1 2 0 2 2 1 0 2 2 2 2 0 2 0 2 0 0 0 2 0 0 0 2 2 0 2 0 1 0 0 0 1 2 1 2 2 2 1 0 1 0 1 2 2 2 2 0 1 1 0 2 2 2 1 1 2 2 0 1 0 1 2 1 2 0 0 1 2 0 2 0 2 0 2 2 2 0 1 0 2 2 1 2 0 2 0 2 1 2 0 1 2 1 1 ...
output:
304775
result:
wrong answer 1st numbers differ - expected: '206384', found: '304775'