CF708C Centroids（树形DP）

Description

题目描述

我们定义一个点x是整棵树的重心（总共有n个点）当且仅当删掉点x后所有连通块的大小都不超过n/2。我们定义一次操作为删掉一条边再加上一条边，必须满足操作后整个图还是一棵树。给你一棵n个点的树，求每个点能否在一次操作后成为重心。

输入

第一行一个整数n；

接下来n-1行每行两个整数u，v：表示第u个点与第v个点间有一条边。

输出

共1行n个数：第i个数表示第i个点能否在一次操作后成为重心。（若可以则输出1，若不可以则输出0）

样例输入

3

1 2

2 3

样例输出

1 1 1

样例解释
样例1：我们可以删掉边(2,3)，然后加上边(1,3)，这样第一个点就可以成为重心。

树形DP

这题处理的有点复杂，但是写出来程序还是很清晰的。

我们考虑一下如何处理这个问题：（原树是以1为根）

首先，如果一个点在原树中本来就是重心，那我们就不用动他。

另一种情况就是：

这个点在树中并不是重心，那一棵$size$大于$n/2$有两种可能：

（1）那棵不符合条件的树在这个点的子树中

那我们看他能不能一次操作变成重心就是看他从这棵子树中最多能分出来的节点（$<=n/2$），分出来之后是否小于n/2。

（2）那棵不符合条件的树不在这个点的子树中

那我们就需要求从父亲这个树（$n-size[i]$）上能最多分出多少个节点（$<=n/2$），像（1）一样做就可以了。

如何做到？

这里就有点繁琐。

$sons:$子树中节点数最多的儿子是哪个

$fas:$父亲这个树（$n-size[i]$）上能最多分出多少个节点（$<=n/2$）

$st:$从子树上分出最多的节点（$<=n/2$）

$nd:$从子树上分出第二多的节点（$<=n/2$）

$id:$st是在以哪一个点为根的子树上的

通过这五个数组我们就能知道答案。

除了$fas$,其他的都比较好求，但是$fas$我们要单独一遍$dfs$求得。

$sons$也可以用作判断$n/2$的子树在$father$中还是在$son$中。

第一遍dfs:

void dfs(int u,int faa)
{
	size[u]=1;
	sons[u]=0;
	for(int i=head[u];i;i=nxt[i])
	{
		int v=to[i];
		if(v!=faa)
		{
			fa[v]=u;
			dfs(v,u);
			size[u]+=size[v];//size
			if(size[v]>size[sons[u]])//sons
			{
				sons[u]=v;
			}
			if(size[v]<=n/2)//st,nd
			{
				if(size[v]>st[u])
				{
					nd[u]=st[u];
					st[u]=size[v];
					id[u]=v;
				}else{
					if(size[v]>nd[u])
					{
						nd[u]=size[v];
					}
				}
			}else{
				if(st[v]>st[u])
				{
					nd[u]=st[u];
					st[u]=st[v];
					id[u]=v;
				}else{
					if(st[v]>nd[u])
					{
						nd[u]=st[v];
					}
				}
			}
		}
	}
	if(n-size[u]>size[sons[u]])//sons
	{
		sons[u]=faa;
	}
}

第二遍dfs:

void dfs1(int u,int fa)
{
	for(int i=head[u];i;i=nxt[i])
	{
		int v=to[i];
		if(v!=fa)//fas
		{
			if(n-size[v]<=n/2)
			{
				fas[v]=max(fas[v],n-size[v]);
			}
			fas[v]=max(fas[v],fas[u]);
			if(id[u]==v)
			{
				fas[v]=max(fas[v],nd[u]);
			}else{
				fas[v]=max(fas[v],st[u]);
			}
			dfs1(v,u);
		}
	}
}

判断：
如果是n/2不在子树中：

那么我们只要$n-size[i]-fas[i]<=n/2$就可以（把分出来的子树接到根节点上）。

如果是n/2在子树中：

那么我们只要$size[sons[i]]-st[i]<=n/2$就可以（同理）。

#include<bits/stdc++.h>
#define N 400010
using namespace std;
int to[N<<1],nxt[N<<1],head[N],size[N],sons[N],fa[N],fas[N],cnt,n,x,y,st[N],nd[N],id[N];
void adde(int x,int y)
{
	to[++cnt]=y;
	nxt[cnt]=head[x];
	head[x]=cnt;
}
void dfs(int u,int faa)
{
	size[u]=1;
	sons[u]=0;
	for(int i=head[u];i;i=nxt[i])
	{
		int v=to[i];
		if(v!=faa)
		{
			fa[v]=u;
			dfs(v,u);
			size[u]+=size[v];//size
			if(size[v]>size[sons[u]])//sons
			{
				sons[u]=v;
			}
			if(size[v]<=n/2)//st,nd
			{
				if(size[v]>st[u])
				{
					nd[u]=st[u];
					st[u]=size[v];
					id[u]=v;
				}else{
					if(size[v]>nd[u])
					{
						nd[u]=size[v];
					}
				}
			}else{
				if(st[v]>st[u])
				{
					nd[u]=st[u];
					st[u]=st[v];
					id[u]=v;
				}else{
					if(st[v]>nd[u])
					{
						nd[u]=st[v];
					}
				}
			}
		}
	}
	if(n-size[u]>size[sons[u]])//sons
	{
		sons[u]=faa;
	}
}
void dfs1(int u,int fa)
{
	for(int i=head[u];i;i=nxt[i])
	{
		int v=to[i];
		if(v!=fa)//fas
		{
			if(n-size[v]<=n/2)
			{
				fas[v]=max(fas[v],n-size[v]);
			}
			fas[v]=max(fas[v],fas[u]);
			if(id[u]==v)
			{
				fas[v]=max(fas[v],nd[u]);
			}else{
				fas[v]=max(fas[v],st[u]);
			}
			dfs1(v,u);
		}
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		adde(x,y);
		adde(y,x); 
	}
	dfs(1,-1);
	dfs1(1,-1);
	for(int i=1;i<=n;i++)
	{
		if(sons[i]==fa[i])
		{
			printf("%d ",n-size[i]-fas[i]<=n/2);
		}else{
			printf("%d ",size[sons[i]]-st[i]<=n/2);
		}
	}
	return 0;
}