| 18774 | 2025-11-04 17:46:34 | algopro | Fire on a Tree | cpp17 | Wrong answer 0/100 | 425ms | 42160 KiB |
// UUID: 5b5ea352-8557-4448-8056-78c40a40e783
#include <bits/stdc++.h>
using namespace std;
int n, m;
vector<vector<int>> adj, dpv;
vector<int> ledp, dp, dpvsum;
void dfs1(int v, int p) {
int maxi = 0;
for (int u : adj[v]) {
if (u==p)continue;
dfs1(u, v);
maxi=max(maxi, ledp[u]);
ledp[v]+=ledp[u];
}
ledp[v]++;
ledp[v]-=maxi;
return;
}
void dfs2(int v, int p) {
dpvsum[v]++;
for (int u : adj[v]) {
if (u==p) {
if (adj[u].size()<=2) {
dpv[v].push_back(1);
dpvsum[v]++;
} else {
if (dpv[u][0]==ledp[v]||dpv[u][1]==ledp[v]) {
dpv[v].push_back(dpv[u][2]);
dpvsum[v]+=dpv[u][2];
} else {
dpv[v].push_back(dpv[u][1]);
dpvsum[v]+=dpv[u][1];
}
}
} else {
dpv[v].push_back(ledp[u]);
dpvsum[v]+=ledp[u];
}
}
sort(dpv[v].rbegin(), dpv[v].rend());
dp[v]=dpvsum[v]-dpv[v][0];
for (int u : adj[v]) {
if (u==p)continue;
dfs2(u, v);
}
}
int main() {
cin >> n;
adj.assign(n+1, {});
dpv.assign(n+1, {});
dp.assign(n+1, 0);
ledp.assign(n+1, 0);
dpvsum.assign(n+1, 0);
for (int i = 1; i < n; i++) {
int a, b;
cin >> a >> b;
adj[a+1].push_back(b+1);
adj[b+1].push_back(a+1);
}
dfs1(2, 0);
dfs2(2, 0);
for (int i = 1; i <= n; i++)cout<<dp[i]<<' ';
cout<<'\n';
}
| Subtask | Sum | Test | Verdict | Time | Memory | ||
|---|---|---|---|---|---|---|---|
| subtask1 | 0/0 | ||||||
| 1 | Accepted | 1ms | 316 KiB | ||||
| 2 | Accepted | 1ms | 316 KiB | ||||
| subtask2 | 0/35 | ||||||
| 3 | Wrong answer | 1ms | 316 KiB | ||||
| 4 | Accepted | 1ms | 316 KiB | ||||
| 5 | Accepted | 1ms | 316 KiB | ||||
| 6 | Wrong answer | 1ms | 512 KiB | ||||
| 7 | Wrong answer | 1ms | 316 KiB | ||||
| subtask3 | 0/25 | ||||||
| 8 | Wrong answer | 1ms | 500 KiB | ||||
| 9 | Wrong answer | 3ms | 564 KiB | ||||
| 10 | Accepted | 3ms | 820 KiB | ||||
| 11 | Wrong answer | 3ms | 564 KiB | ||||
| 12 | Wrong answer | 3ms | 564 KiB | ||||
| 13 | Wrong answer | 3ms | 564 KiB | ||||
| subtask4 | 0/40 | ||||||
| 14 | Wrong answer | 190ms | 14756 KiB | ||||
| 15 | Wrong answer | 284ms | 20332 KiB | ||||
| 16 | Accepted | 347ms | 42160 KiB | ||||
| 17 | Wrong answer | 319ms | 25900 KiB | ||||
| 18 | Wrong answer | 301ms | 23460 KiB | ||||
| 19 | Wrong answer | 337ms | 24884 KiB | ||||
| 20 | Wrong answer | 384ms | 25908 KiB | ||||
| 21 | Wrong answer | 425ms | 25292 KiB | ||||
| 22 | Wrong answer | 319ms | 24884 KiB | ||||