[CodeForces 1009F]Dominant Indices(dsu on tree)

题意
给定一棵树（以1为根），对于每个结点，统计其子树（包括其自身）中所有结点到达该结点的距离。定义节点数最多的距离为这个结点的dominant index，若多个距离的结点数相等则取最小的距离，对编号1~n的结点，顺次输出每个结点的dominant index

思路
dsu on tree，本来打算在记录子树时用depth[]数组记录所有深度，然后对重链上推的时候把所有depth[]也暴力往后推1，结果tle on 131

后来改了做法，计算size的时候顺便计算每个点的深度（相对根结点的深度），然后直接记录子树中每个深度的点数，用子树树根结点深度减去点数最多的深度就是答案了，上推的时候就可以直接上推了。

AC代码

#include<bits/stdc++.h>
#include<math.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+5;
int n,ma,mo;
int h[maxn],p[maxn*2],nxt[maxn*2];
long long ans[maxn],siz[maxn],son[maxn],depth[maxn],de[maxn];//depth对深度的计数 
void add(int now,int f,int val)
{
	for(int i=h[now];i;i=nxt[i])
	{
		if(p[i]!=f&&!son[p[i]]) add(p[i],now,val);//不计算被标记为重儿子的子树（因为已经计算过了） 
	}
	depth[de[now]]+=val;
	if(val+1)
	{
		if((ma==-1)||depth[de[now]]>depth[ma]||(depth[de[now]]==depth[ma]&&de[now]<ma))ma=de[now];
	}
	return;
}
int sea(int now,int f)//预处理，计算每个子树的大小 
{
	de[now]=de[f]+1;
	int mx,son_;
	siz[now]=1;
	for(int i=h[now];i;i=nxt[i])
	if(p[i]!=f)
	{
		sea(p[i],now);
		siz[now]+=siz[p[i]];
	}
	return 0;	
}
int dfs(int now,int f,int keep)//keep标记用来说明是否要保留影响 
{
	int son_,mx;
	mx=son_=-1;
	for(int i=h[now];i;i=nxt[i])
	if(p[i]!=f)if(siz[p[i]]>mx)mx=siz[p[i]],son_=p[i];//记录重儿子 
	for(int i=h[now];i;i=nxt[i])
	if(p[i]!=f&&p[i]!=son_)
	{
		dfs(p[i],now,0);//计算所有非重儿子的子树，清除影响 
	}
	if(son_!=-1)
	{son[son_]=1;dfs(son_,now,1);}//标记重儿子，计算重儿子子树，不清除影响 
	add(now,f,1);//这里加入所有非重儿子的子树，在这个函数里检查标记看是否为重儿子，是则不计算对应子树 
	if(son_!=-1)son[son_]=0;//清除标记，为后面的清理做准备（如果keep为1也可以不清） 
	ans[now]=ma-de[now];
	if(!keep)//不保留影响则清除，此时重儿子标记已经删除，所以会清理整棵子树的影响 
	add(now,f,-1),ma=-1,mo=0;
}
int main()
{
	scanf("%d",&n);ma=-1;
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		p[i*2-1]=y;nxt[i*2-1]=h[x];h[x]=i*2-1;
		p[i*2]=x;nxt[i*2]=h[y];h[y]=i*2;
	}
	sea(1,0);
	dfs(1,0,1);
	for(int i=1;i<=n;i++)printf("%lld\n",ans[i]);
	return 0;
}