187602025-11-04 17:06:26algoproFire on a Treecpp17Wrong answer 0/100296ms40572 KiB
// UUID: c1a0dd21-9969-4f5b-8808-7a9902dff98f
#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+1;
		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
1Accepted6ms4916 KiB
2Accepted4ms4916 KiB
subtask20/35
3Wrong answer6ms5044 KiB
4Wrong answer6ms5144 KiB
5Accepted6ms5036 KiB
6Wrong answer4ms4916 KiB
7Wrong answer4ms4916 KiB
subtask30/25
8Wrong answer4ms5032 KiB
9Wrong answer6ms5176 KiB
10Accepted6ms5428 KiB
11Wrong answer7ms5176 KiB
12Wrong answer7ms5356 KiB
13Wrong answer6ms5212 KiB
subtask40/40
14Wrong answer101ms9868 KiB
15Wrong answer172ms12188 KiB
16Accepted201ms40572 KiB
17Wrong answer247ms14080 KiB
18Wrong answer190ms12696 KiB
19Wrong answer296ms13356 KiB
20Wrong answer270ms14216 KiB
21Wrong answer187ms13364 KiB
22Wrong answer187ms13204 KiB