UOJ #139.【UER #4】被删除的黑白树 长链剖分

题意

给出一棵有根树，要求对尽量多的节点染色，使得每个叶节点到根的路径上的染色节点数都相等。
$n\le 10^5$

分析

开始刷水题了。。。
设dp[i,j]表示以i为根的子树中每个叶节点到i路径上的染色点数恰好为j时的答案，用长链剖分优化转移就是$O(n)$的了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>

const int N=100005;

int n,cnt,last[N],nx[N],dep[N],son[N],h[N],t[N],w[N],sz;
struct edge{int to,next;}e[N*2];

void addedge(int u,int v)
{
	e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
	e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}

void dfs1(int x,int fa)
{
	for (int i=last[x];i;i=e[i].next)
	{
		if (e[i].to==fa) continue;
		dfs1(e[i].to,x);
		if (dep[e[i].to]>dep[son[x]]) son[x]=e[i].to;
	}
	dep[x]=dep[son[x]]+1;
}

void dfs2(int x,int fa)
{
	if (son[x])
	{
		dfs2(son[x],x);
		h[x]=h[son[x]];t[x]=t[son[x]];
	}
	for (int i=last[x];i;i=e[i].next)
	{
		if (e[i].to==fa||e[i].to==son[x]) continue;
		dfs2(e[i].to,x);
		int k=h[x],ls=0;
		for (int j=h[e[i].to];j&&k;j=nx[j],ls=k,k=nx[k]) w[k]+=w[j];
		nx[ls]=0;t[x]=ls;
	}
	if (son[x]) nx[t[x]]=++sz,w[sz]=w[t[x]]+1,t[x]=sz;
	else h[x]=t[x]=++sz,w[sz]=1;
}

int main()
{
	scanf("%d",&n);
	for (int i=1;i<n;i++)
	{
		int x,y;scanf("%d%d",&x,&y);
		addedge(x,y);
	}
	dfs1(1,0);
	dfs2(1,0);
	int mx=0;
	for (int i=h[1];i;i=nx[i]) mx=std::max(mx,w[i]);
	printf("%d\n",mx);
	return 0;
}