CodeTON Round 1 (Div. 1 + Div. 2) E. Equal Tree Sums题解

题外话

老年选手掉分的日常

题目大意

给一棵无根树的每个节点赋一个权值，使得删除树中任意一个节点后，剩下的连通块的点权和相等

题解

假设无根树的根为1号点

设节点u权值为a[u]，子树权值和为sum[u]

u的某一个儿子为v，v的某一个儿子为w，u的儿子总数为son

那么由条件知

sum[1]-sum[u]=sum[v] (u!=1)

移项可得：

sum[1]=sum[u]+sum[v]

同理可得

sum[1]=sum[v]+sum[w]

所以有sum[u]+sum[v]=sum[v]+sum[w]，即sum[u]=sum[w]

意思就是说爷爷的sum和孙子的sum值是一样的 (除1号节点外)

如果我们设1号节点的儿子sum值为a，孙子sum值为b

（注意这里的a是一个常数与权值a[u]不一样）

那么sum[1]=a+b

不妨设sum[u]=a，那么sum[v]=b

又因为sum[u]=a[u]+Σsum[v]=a[u]+son*sum[v]

所以a=a[u]+son*b

即a[u]=a-son*b

考虑sum[1]=a[1] + (1号节点儿子总数)*a，sum[1]=a+b

所以a[1]=b-(1号节点儿子总数)*a

由于-10^5<=a[x]<=10^5 且a[x]!=0

我们可以“巧妙地” 令a=-1 , b=1

于是我们发现a[u]= -1-son = - u节点度数

a[1]=1-(1号节点儿子总数)*(-1) = 1号节点度数

由于我们之前是“不妨设sum[u]=a”

所以对于一般的u(u!=1)，a[u]=u节点度数*（-1）^(u的在树中的深度)

也就是官方题解中所说的二染色后一种颜色取+deg[u]，另一种颜色取-deg[u]

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
inline int gi()
{
	char c;int num=0,flg=1;
	while((c=getchar())<'0'||c>'9')if(c=='-')flg=-1;
	while(c>='0'&&c<='9'){num=num*10+c-48;c=getchar();}
	return num*flg;
}
#define N 100005
int fir[N],to[2*N],nxt[2*N],cnt;
void adde(int a,int b){to[++cnt]=b;nxt[cnt]=fir[a];fir[a]=cnt;}
int d[N];
void dfs(int u,int fa,int flg)
{
	d[u]*=flg;
	for(int v,p=fir[u];p;p=nxt[p]){
		v=to[p];
		if(v!=fa)
			dfs(v,u,-flg);
	}
}
int main()
{
	int T,n,i,u,v;
	T=gi();
	while(T--){
		n=gi();cnt=0;
		for(i=1;i<=n;i++)d[i]=fir[i]=0;
		for(i=1;i<n;i++){
			u=gi();v=gi();
			adde(u,v);adde(v,u);
			d[u]++;d[v]++;
		}
		dfs(1,0,1);
		for(i=1;i<=n;i++)
			printf("%d ",d[i]);
		printf("\n");
	}
}