标签：CF708C int siz up Centroids wgt dw1 mor

确实不简单。

首先要理解题意。你可以拿掉任意一棵子树，接到一个节点上，但是拿掉的子树的大小并不是1~siz[u]。假设树根为root，我们找到子树中最大的子树，从中拿掉最大的小子树，然后接到根上，
如果这个时候符合题意，那么以root为根就可以通过改造成为树的重心。

问题的关键就是求出以root为根时，root的每个儿子能拿掉的子树大小的最大值是多少。可以用树形DP，算两次。

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

const int N = 4e5 + 100; 

int head[N], cnt, siz[N], wgt[N];
int n;
int dw1[N], dw2[N], p1[N], up[N], mor[N], idx[N];//mor[u]：以u为根的子树中，大于 n / 2的子树有多少个 
bool ans[N]; //idx[u]最大子树的编号 

struct Edge {
	int v, nxt;
} e[N <<1];

void AddEdge(int u, int v) {
	e[++cnt].v = v;
	e[cnt].nxt = head[u];
	head[u] = cnt;
}

void update(int u, int v, int val) {
	if( val > n / 2) {
		mor[u] ++;
		return;
	}
	if( val >= dw1[u]) {
		dw2[u] = dw1[u];
		p1[u] = v;
		dw1[u] = val;
	}
	else if( val > dw2[u])
		dw2[u] = val;
} 

void dfs1(int u, int fa) {
	siz[u] = 1;
	int wgt = 0;
	for(int i = head[u]; i; i = e[i].nxt) {
		int v = e[i].v;
		if( v == fa)	continue;
		dfs1(v, u);
		siz[u] += siz[v];//dw1[u]:以u为根的子树中，满足siz <= n / 2的最大值  
		if(siz[v] <= n / 2)
	    	update(u, v, max(dw1[v], siz[v]));//!!!
	    else
	        update(u, v, dw1[v]);
		if( siz[v] > wgt) 
			idx[u] = v, wgt = siz[v];
		if( siz[v] > n / 2)
			mor[u] ++;
	}
	if( n - siz[u] <= n / 2) {
		up[u] = n - siz[u];
	}
	else 
		mor[u] ++;
	if( n - siz[u] > wgt) 
		idx[u] = n + 1;//上方子树 
}

void dfs2(int u, int fa) {
	for(int i = head[u]; i; i = e[i].nxt) {
		int v = e[i].v;
		if( v == fa)	continue;
/*		if(v == 6) {
			printf("D: %d %d\n", dw1[u], p1[u]); 
		} */
		if(p1[u] != v) {
			up[v] = max(up[v], max(dw1[u], up[u]));
		}
		else up[v] = max(up[v], max(dw2[u], up[u]));
		dfs2(v, u);
	}
	if(mor[u] > 1)	ans[u] = false;
	else if( mor[u] == 0)  ans[u] = true; 
	else {
		if( idx[u] == n + 1) {
			ans[u] = ( (n - siz[u]) - up[u]) <= n / 2;
		}
		else {
			ans[u] = (siz[idx[u]] - dw1[u]) <= n / 2;
		}
	} 
}

int main()
{
//	freopen("E:\\data.in.txt", "r", stdin);
	scanf("%d", &n);
	for(int i = 1; i < n; i ++) {
		int u, v;
		scanf("%d%d", &u, &v);
		AddEdge(u, v), AddEdge(v, u);
	}
	dfs1(1, 0);
	dfs2(1, 0);
	for(int i = 1; i <= n; i ++) {
		printf("%d ", ans[i]); 
	} 
	return 0;
}

标签：CF708C,int,siz,up,Centroids,wgt,dw1,mor
来源： https://www.cnblogs.com/wyy0804/p/13736735.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。