原文链接https://www.cnblogs.com/zhouzhendong/p/UOJ345.html
前言
我真的是越来越菜了，连树形DP都感觉陌生了。
题解
首先，我们来看看在不断生长叶子会发生什么。
第一种：顺着生长方向走。
第二种：在某一个节点的某些子树依次生长，达到他们之间互相消耗的作用。
对于一个子树 x，假设初始情况下，树心在点 x 上，那么在 x 子树中生长若干次之后，树心离开 x，那么我们考虑求出在树心离开 x 之后，能将 x 往 x 子树方向拉的次数的上界 Max[x] 和下界 Min[x] 。
设 size[x] 表示 x 子树的大小。
显然 Max[x] = size[x]-1 。
而 Min[x] 怎么求？
如果 x 是叶子，那么显然 Min[x] = 0； 如果 x 不是，那么设 x 的儿子中 Max 值最大的节点为 y ，设 $s = sum_{i eq y} (Max[i]+1)$ 如果 $s geq Min[y]+1$ ，则 Min[x] = (s-(Min[y]+1)) mod 2 ，否则 Min[x] = Min[y]+1 - s 。
于是我们已经可以得到根节点的答案了。
那么其他节点呢？
其实大同小异，类似于换根dp地从根出发 dp 一边就好了。
代码
#include <bits/stdc++.h>
#define clr(x) memset(x,0,sizeof (x))
#define For(i,a,b) for (int i=a;i<=b;i++)
#define Fod(i,b,a) for (int i=b;i>=a;i--)
#define pb(x) push_back(x)
#define mp(x,y) make_pair(x,y)
#define fi first
#define se second
#define real __zzd001
#define _SEED_ ('C'+'L'+'Y'+'A'+'K'+'I'+'O'+'I')
#define outval(x) printf(#x" = %d
",x)
#define outvec(x) printf("vec "#x" = ");for (auto _v : x)printf("%d ",_v);puts("")
#define outtag(x) puts("----------"#x"----------")
#define outarr(a,L,R) printf(#a"[%d...%d] = ",L,R);
For(_v2,L,R)printf("%d ",a[_v2]);puts("");
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef vector <int> vi;
LL read(){
LL x=0,f=0;
char ch=getchar();
while (!isdigit(ch))
f|=ch=='-',ch=getchar();
while (isdigit(ch))
x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
return f?-x:x;
}
const int N=100005;
int W,T,n;
vector <int> e[N];
int size[N],Min[N];
int fa[N];
void delfa(int x,int pre){
size[x]=1,fa[x]=pre;
for (auto y : e[x])
if (y!=pre)
delfa(y,x),size[x]+=size[y];
for (int i=0;i<(int)e[x].size();i++)
if (e[x][i]==pre){
swap(e[x][i],e[x].back());
e[x].pop_back();
break;
}
}
void dfs(int x){
for (auto y : e[x])
dfs(y);
if (e[x].size()>0){
int s=size[x]-1,mx=0,mi=-1;
for (auto y : e[x])
if (mx<=size[y])
mx=size[y],mi=Min[y]+1;
s-=mx;
if (s>=mi)
Min[x]=(s-mi)&1;
else
Min[x]=mi-s;
}
else
Min[x]=0;
}
int res[N];
void solve(int x,int d,int s=0){
int fsize=n-size[x]-d;
if (max(Min[x],s)<=min(size[x]-1,fsize))
res[x]=1;
else
res[x]=0;
if (e[x].empty())
return;
int len=e[x].size()+1;
vector <int> sum(len,0),mi(len,-1),mx(len,0);
sum[0]=fsize,mx[0]=fsize,mi[0]=s;
for (int i=1;i<len;i++){
int y=e[x][i-1];
sum[i]=sum[i-1]+size[y];
if (mx[i-1]<=size[y])
mx[i]=size[y],mi[i]=Min[y]+1;
else
mx[i]=mx[i-1],mi[i]=mi[i-1];
}
int sumR=0,miR=-1,mxR=0;
for (int i=len-1;i>=1;i--){
int y=e[x][i-1];
int Sum=sum[i-1]+sumR;
int Mx=mxR,Mi=miR;
if (Mx<=mx[i-1])
Mx=mx[i-1],Mi=mi[i-1];
Sum-=Mx;
solve(y,d+1,Sum>=Mi?((Sum-Mi)&1):Mi-Sum);
sumR+=size[y];
if (mxR<=size[y])
mxR=size[y],miR=Min[y]+1;
}
}
void chk(int x,int v){
res[x]&=v;
for (auto y : e[x])
chk(y,v^1);
}
void Solve(){
n=read();
For(i,0,n)
e[i].clear();
For(i,1,n-1){
int x=read(),y=read();
e[x].pb(y),e[y].pb(x);
}
delfa(1,0);
dfs(1);
solve(1,0);
chk(1,n&1);
if (W==3)
printf("%d
",res[1]);
else {
for (int i=1;i<=n;i++)
putchar('0'+res[i]);
puts("");
}
}
int main(){
W=read(),T=read();
while (T--)
Solve();
return 0;
}