#include <bits/stdc++.h>
using namespace std;
const int N = 1E5 + 5, M = 2E5 + 5;
int n, m, q, d, u, f[N], g[N], ue[N], ve[N], lz[N], col[N];
int que[M];
bool chk[N], mod[N];
vector<int> adj[N], com[N];
void init() {
for (int i = 1; i <= n; i++) {
lz[i] = col[i] = 0;
mod[i] = false;
adj[i].clear(); com[i].clear();
}
}
void DFS(int u, int p = 0) {
col[u] = col[p];
for (int &v : adj[u]) {
int c = u ^ ue[v] ^ ve[v];
if (c != p) {
DFS(c, u);
}
}
}
void update(int u, int dif, int p = 0) {
f[u] += dif; lz[u] += dif;
for (int &v : com[u]) {
int c = u ^ col[ue[v]] ^ col[ve[v]];
if (!chk[v]) {
g[v] += dif;
} else if (c != p) {
update(c, dif, u);
}
}
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
cin >> n >> m >> q;
fill(f + 1, f + n + 1, 1);
for (int i = 1; i < n; i++) {
cin >> ue[i] >> ve[i];
g[i] = 2;
}
for (int i = 1; i <= m; i++) {
cin >> que[i];
}
d = sqrt(m / 4 + 1);
for (int bl = 1; bl <= m; bl += d) {
init();
for (int i = 0; i < d && bl + i <= m; i++) {
mod[que[bl + i]] = true;
}
for (int i = 1; i < n; i++) {
if (!mod[i] && chk[i]) {
adj[ue[i]].push_back(i);
adj[ve[i]].push_back(i);
}
}
for (int i = 1; i <= n; i++) {
if (col[i] == 0) {
col[0] = i;
DFS(i);
}
}
for (int i = 1; i < n; i++) {
if (mod[i]) {
com[col[ue[i]]].push_back(i);
com[col[ve[i]]].push_back(i);
}
}
for (int i = 0; i < d && bl + i <= m; i++) {
int e = que[bl + i];
if (!chk[e]) {
int sum = (f[col[ue[e]]] + f[col[ve[e]]] + g[e]) / 2;
update(col[ue[e]], sum - f[col[ue[e]]]);
update(col[ve[e]], sum - f[col[ve[e]]]);
g[e] = 0;
}
chk[e] = !chk[e];
}
for (int i = 1; i <= n; i++) {
f[i] = f[col[i]];
}
for (int i = 1; i < n; i++) {
if (!mod[i] && !chk[i]) {
g[i] += lz[col[ue[i]]] + lz[col[ve[i]]];
}
}
}
while (q--) {
cin >> u;
cout << f[u] << '\n';
}
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
7 ms |
5112 KB |
Output is correct |
2 |
Correct |
7 ms |
5112 KB |
Output is correct |
3 |
Correct |
6 ms |
5084 KB |
Output is correct |
4 |
Correct |
6 ms |
5112 KB |
Output is correct |
5 |
Correct |
6 ms |
5112 KB |
Output is correct |
6 |
Correct |
13 ms |
5112 KB |
Output is correct |
7 |
Correct |
198 ms |
5936 KB |
Output is correct |
8 |
Correct |
197 ms |
6124 KB |
Output is correct |
9 |
Correct |
211 ms |
6048 KB |
Output is correct |
10 |
Execution timed out |
8042 ms |
14340 KB |
Time limit exceeded |
11 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3296 ms |
17244 KB |
Output is correct |
2 |
Correct |
3407 ms |
17204 KB |
Output is correct |
3 |
Correct |
2247 ms |
19064 KB |
Output is correct |
4 |
Correct |
2230 ms |
18980 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
5112 KB |
Output is correct |
2 |
Correct |
6 ms |
5112 KB |
Output is correct |
3 |
Correct |
6 ms |
5112 KB |
Output is correct |
4 |
Correct |
6 ms |
5112 KB |
Output is correct |
5 |
Correct |
6 ms |
5112 KB |
Output is correct |
6 |
Correct |
10 ms |
5240 KB |
Output is correct |
7 |
Correct |
145 ms |
6108 KB |
Output is correct |
8 |
Correct |
5971 ms |
14908 KB |
Output is correct |
9 |
Correct |
7478 ms |
14864 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6953 ms |
14860 KB |
Output is correct |
2 |
Correct |
2250 ms |
19480 KB |
Output is correct |
3 |
Correct |
2299 ms |
19612 KB |
Output is correct |
4 |
Correct |
2294 ms |
19448 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
5112 KB |
Output is correct |
2 |
Correct |
6 ms |
5112 KB |
Output is correct |
3 |
Correct |
6 ms |
5112 KB |
Output is correct |
4 |
Correct |
6 ms |
5112 KB |
Output is correct |
5 |
Correct |
11 ms |
5112 KB |
Output is correct |
6 |
Correct |
200 ms |
6008 KB |
Output is correct |
7 |
Execution timed out |
8071 ms |
14212 KB |
Time limit exceeded |
8 |
Halted |
0 ms |
0 KB |
- |