187642025-11-04 17:08:45horkaFire on a Treecpp17Elfogadva 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";

}
RészfeladatÖsszpontTesztVerdiktIdőMemória
subtask10/0
1Elfogadva4ms4916 KiB
2Elfogadva6ms4916 KiB
subtask235/35
3Elfogadva6ms5116 KiB
4Elfogadva4ms4976 KiB
5Elfogadva4ms4984 KiB
6Elfogadva4ms4964 KiB
7Elfogadva4ms4916 KiB
subtask325/25
8Elfogadva4ms4916 KiB
9Elfogadva7ms5172 KiB
10Elfogadva7ms5428 KiB
11Elfogadva6ms5172 KiB
12Elfogadva7ms5172 KiB
13Elfogadva6ms5172 KiB
subtask440/40
14Elfogadva127ms10036 KiB
15Elfogadva143ms12168 KiB
16Elfogadva193ms40608 KiB
17Elfogadva247ms14068 KiB
18Elfogadva234ms12792 KiB
19Elfogadva218ms13212 KiB
20Elfogadva206ms14204 KiB
21Elfogadva256ms13440 KiB
22Elfogadva202ms13148 KiB