大致题意: 设\(d(x,y)\)表示\(x\)子树内到\(x\)距离为\(y\)的点的个数,对于每个\(x\),求满足\(d(x,y)\)最大的最小的\(y\)。
暴力\(DP\)
首先让我们来思考如何暴力\(DP\)。
这应该还是比较简单的吧。
直接设\(f_{x,i}\)表示在\(x\)的子树内,到\(x\)的距离为\(i\)的点的个数。
则不难推出转移方程:
\[f_{x,0}=1,f_{x,i}=\sum f_{son_x,i-1}
\]
\]
但这样显然跑不过,要优化。
长链剖分
这是一道长链剖分优化\(DP\)的典型例题。
设\(len_x\)为\(x\)到叶节点的最长距离,则不难发现转移方程第二维\(i\)显然不可能超过\(len_x\)。
考虑使用指针(虽然我很不喜欢指针,但貌似这里必须用\(2333\)),对于每条长链顶点给它开一个长度为\(len_x\)的内存供它存储。
然后,对于每条长链的第\(i\)个元素\(t_i\),我们就用第\(i\)个位置作为\(f_{t_i,0}\)。
这样的好处就在于,对于一条长链,我们是可以直接让父节点从子节点那里继承答案的!
是不是非常神奇?
而对于非长儿子,我们暴力合并两条链。
由于每条链只会被合并一次,因此复杂度就达到了无比优秀的\(O(n)\)!
是不是很神奇?
代码
#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define Reg register
#define RI Reg int
#define Con const
#define CI Con int&
#define I inline
#define W while
#define N 1000000
#define add(x,y) (e[++ee].nxt=lnk[x],e[lnk[x]=ee].to=y)
using namespace std;
int n,ee,lnk[N+5];struct edge {int to,nxt;}e[N<<1];
class FastIO
{
private:
#define FS 100000
#define tc() (A==B&&(B=(A=FI)+fread(FI,1,FS,stdin),A==B)?EOF:*A++)
#define pc(c) (C^FS?FO[C++]=c:(fwrite(FO,1,C,stdout),FO[(C=0)++]=c))
#define tn (x<<3)+(x<<1)
#define D isdigit(c=tc())
int T,C;char c,*A,*B,FI[FS],FO[FS],S[FS];
public:
I FastIO() {A=B=FI;}
Tp I void read(Ty& x) {x=0;W(!D);W(x=tn+(c&15),D);}
Ts I void read(Ty& x,Ar&... y) {read(x),read(y...);}
Tp I void write(Ty x) {W(S[++T]=x%10+48,x/=10);W(T) pc(S[T--]);}
Tp I void writeln(Con Ty& x) {write(x),pc('\n');}
I void clear() {fwrite(FO,1,C,stdout),C=0;}
#undef D
}F;
class LongChainDissection//长链剖分优化DP
{
private:
#define Assign(x) (f[x]=p,p+=len[x])//分配内存
#define F5(x,v) ((f[x][ans[x]]<f[x][v]||(f[x][ans[x]]==f[x][v]&&ans[x]>v))&&(ans[x]=v))//更新答案
int son[N+5],len[N+5],ans[N+5],*p,*f[N+5],_f[N+5];
I void dfs(CI x,CI lst)//DFS预处理长儿子与len数组
{
for(RI i=lnk[x];i;i=e[i].nxt) e[i].to^lst&&
(dfs(e[i].to,x),len[e[i].to]>len[son[x]]&&(son[x]=e[i].to));
len[x]=len[son[x]]+1;
}
I void DP(CI x,CI lst)//DP
{
RI i,j;son[x]&&(f[son[x]]=f[x]+1,DP(son[x],x),ans[x]=ans[son[x]]+1);//优先处理长儿子,并继承答案
for(i=lnk[x];i;i=e[i].nxt) if(e[i].to^lst&&e[i].to^son[x])//枚举非长儿子
for(Assign(e[i].to),DP(e[i].to,x),j=1;j<=len[e[i].to];++j)//暴力合并
f[x][j]+=f[e[i].to][j-1],F5(x,j);f[x][0]=1,F5(x,0);
}
public:
I LongChainDissection() {p=_f;}I void Init() {dfs(1,0);}//初始化
I void Solve() {Assign(1),DP(1,0);for(RI i=1;i<=n;++i) F.writeln(ans[i]);}//DP并输出答案
}D;
int main()
{
RI i,x,y;for(F.read(n),i=1;i^n;++i) F.read(x,y),add(x,y),add(y,x);//读入+建边
return D.Init(),D.Solve(),F.clear(),0;//求解
}