| 18779 | 2025-11-04 17:57:42 | CzDani | Fire on a Tree | cpp17 | Accepted 100/100 | 395ms | 45080 KiB |
#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];
dpv[v].push_back(dpvsum[u]-dpv[u][0]-dpv[u][1]);
dpvsum[v]+=dpvsum[u]-dpv[u][0]-dpv[u][1];
} else {
// dpv[v].push_back(dpv[u][1]);
// dpvsum[v]+=dpv[u][1];
dpv[v].push_back(dpvsum[u]-dpv[u][0]-ledp[v]);
dpvsum[v]+=dpvsum[u]-dpv[u][0]-ledp[v];
}
}
} 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 | 35/35 | ||||||
| 3 | Accepted | 1ms | 316 KiB | ||||
| 4 | Accepted | 1ms | 316 KiB | ||||
| 5 | Accepted | 1ms | 316 KiB | ||||
| 6 | Accepted | 1ms | 316 KiB | ||||
| 7 | Accepted | 1ms | 316 KiB | ||||
| subtask3 | 25/25 | ||||||
| 8 | Accepted | 1ms | 316 KiB | ||||
| 9 | Accepted | 2ms | 564 KiB | ||||
| 10 | Accepted | 3ms | 732 KiB | ||||
| 11 | Accepted | 3ms | 564 KiB | ||||
| 12 | Accepted | 3ms | 536 KiB | ||||
| 13 | Accepted | 3ms | 564 KiB | ||||
| subtask4 | 40/40 | ||||||
| 14 | Accepted | 180ms | 14804 KiB | ||||
| 15 | Accepted | 226ms | 20272 KiB | ||||
| 16 | Accepted | 287ms | 45080 KiB | ||||
| 17 | Accepted | 375ms | 25908 KiB | ||||
| 18 | Accepted | 370ms | 23604 KiB | ||||
| 19 | Accepted | 328ms | 24884 KiB | ||||
| 20 | Accepted | 314ms | 25904 KiB | ||||
| 21 | Accepted | 395ms | 25396 KiB | ||||
| 22 | Accepted | 300ms | 24880 KiB | ||||