187642025-11-04 17:08:45horkaFire on a Treecpp17Accepted 100/100256ms40608 KiB
#include <bits/stdc++.h>
using namespace std;
const int c=2e5+10;
vector<int> adj[c];
int dp[c];
void dfs1(int cs, int p)
{
	dp[cs]=1;
	int s=0,mx=0;
	for(int &i:adj[cs])
		if(i!=p)
		{
			dfs1(i,cs);
			s+=dp[i];
			mx=max(mx,dp[i]);
		}
	dp[cs]+=s-mx;
}
int ans[c];
void dfs2(int cs, int p, int ert)
{
	int s=ert,mx=ert;
	for(int &i:adj[cs])
		if(i!=p) s+=dp[i],mx=max(mx,dp[i]);
	ans[cs]=1+s-mx;
	vector<array<int, 2>> gy{{ert,p}};
	for(int &i:adj[cs])
		if(i!=p) gy.push_back({dp[i],i});
	sort(gy.rbegin(),gy.rend());
	for(int &i:adj[cs])
	{
		if(i==p) continue;
		int tmp=s-dp[i];
		if(i==gy[0][1]) tmp-=gy[1][0];
		else tmp-=gy[0][0];
		dfs2(i,cs,tmp+1);
	}
}
int main() {
	ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	int n; cin>>n;
	for(int i=1; i<n; i++)
	{
		int a,b; cin>>a>>b;
		a++,b++;
		adj[a].push_back(b);
		adj[b].push_back(a);
	}
	dfs1(1,0);
	dfs2(1,0,0);
	for(int i=1; i<=n; i++)
		cout<<ans[i]<<" ";
	cout<<"\n";

}
SubtaskSumTestVerdictTimeMemory
subtask10/0
1Accepted4ms4916 KiB
2Accepted6ms4916 KiB
subtask235/35
3Accepted6ms5116 KiB
4Accepted4ms4976 KiB
5Accepted4ms4984 KiB
6Accepted4ms4964 KiB
7Accepted4ms4916 KiB
subtask325/25
8Accepted4ms4916 KiB
9Accepted7ms5172 KiB
10Accepted7ms5428 KiB
11Accepted6ms5172 KiB
12Accepted7ms5172 KiB
13Accepted6ms5172 KiB
subtask440/40
14Accepted127ms10036 KiB
15Accepted143ms12168 KiB
16Accepted193ms40608 KiB
17Accepted247ms14068 KiB
18Accepted234ms12792 KiB
19Accepted218ms13212 KiB
20Accepted206ms14204 KiB
21Accepted256ms13440 KiB
22Accepted202ms13148 KiB