Educational Codeforces Round 47 (Rated for Div. 2) F. Dominant Indices（长链剖分+树上dp）

F. Dominant Indices

题意

给定一棵根为 1 的树， 对于每个节点，求该结点的子树中，哪个距离下的节点数量最多。若有多种情况，输出最小的值。

分析

长链剖分+树上dp
首先对这课树进行长链剖分，用sum[x]指针数组表示节点x不同距离的点的数量。在同一条长链中，由于sum[son][i]=sum[father][i+1]，因此sum[son]=sum[father]+1。只需在两条链合并的时候，进行dp操作即可。
对于答案的记录，存放在res数组中，每次算好儿子的答案后，将自己的答案更新。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 1e6+10;
int n,m,tot;
int head[N];
struct node{
	int v,ne;
}edge[N<<1];
void add(int x,int y)
{
	edge[++tot].v=y;
	edge[tot].ne=head[x];
	head[x]=tot;
}
//长链剖分 
int deep[N],size[N],son[N];
//deep记录该结点深度,h_size记录重链长度,son记录重儿子(深度最深)
void dfs1(int now,int father)
{
	deep[now]=deep[father]+1;
	size[now]=deep[now];
	for(int i=head[now];i;i=edge[i].ne)
	{
		if(edge[i].v==father)
		continue;
		dfs1(edge[i].v,now);
		size[now]=max(size[now],size[edge[i].v]);
		if(size[edge[i].v]>size[son[now]])
		son[now]=edge[i].v;  //长度最大的为重儿子
	}
}
int *sum[N],tmp[N],*id=tmp;
//sum为指针数组,数组总大小为maxn(所有长链加起来为n个点),id为指针
int res[N];
void dfs(int now,int father)
{
	sum[now][0]=1;//每一条链中距离为0的点只有一个 
	if(son[now])//遍历长链 
	{
		sum[son[now]]=sum[now]+1;//更新链 
		dfs(son[now],now);
		res[now]=res[son[now]]+1;//向上回溯 
	}
	for(int i=head[now];i;i=edge[i].ne)//寻找下一个长链 
	{
		if(edge[i].v==father||edge[i].v==son[now])
		continue;
		sum[edge[i].v]=id;//edge[i].v为下一个长链的top 
		int len=size[edge[i].v]-deep[edge[i].v]+1;//链长度 
		id+=len;
		dfs(edge[i].v,now);//更新每个点 
		for(int j=1;j<=len;j++)
		{
			sum[now][j]+=sum[edge[i].v][j-1];//两条链合并 
			if(sum[now][j]>sum[now][res[now]]||(sum[now][j]==sum[now][res[now]]&&res[now]>j))
			res[now]=j;//res记录的是答案 
		}
	}
	if(sum[now][res[now]]==1)//满足题意，输出较小的值 
	res[now]=0; 
}
int main()
{
	scanf("%d",&n);
	int x,y;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	dfs1(1,0);
	sum[1]=id;id+=size[1];
	dfs(1,0);
	for(int i=1;i<=n;i++)
	printf("%d\n",res[i]);
}