187632025-11-04 17:08:25algoproFire on a Treecpp17Accepted 100/100268ms40536 KiB
// UUID: d6e0bfe5-dd0f-42aa-bb05-bba2b0b3aa4a
#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
1Accepted4ms5108 KiB
2Accepted4ms4916 KiB
subtask235/35
3Accepted4ms4916 KiB
4Accepted6ms4916 KiB
5Accepted6ms4916 KiB
6Accepted6ms4916 KiB
7Accepted4ms4916 KiB
subtask325/25
8Accepted4ms4916 KiB
9Accepted4ms4952 KiB
10Accepted7ms5488 KiB
11Accepted7ms5172 KiB
12Accepted7ms5216 KiB
13Accepted6ms5220 KiB
subtask440/40
14Accepted94ms9984 KiB
15Accepted171ms12124 KiB
16Accepted188ms40536 KiB
17Accepted246ms14188 KiB
18Accepted180ms12852 KiB
19Accepted268ms13108 KiB
20Accepted194ms14300 KiB
21Accepted192ms13432 KiB
22Accepted254ms13108 KiB