【Codeforces1009F】—Dominant Indices（长链剖分）

传送门

题意：给你一棵树，对每个点询问子树中节点数目最多的深度

长链剖分板子题
不多解释

#include<bits/stdc++.h>
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) res=(res+(res<<2)<<1)+(ch^48),ch=getchar();
	return res*f;
}
const int N=1e6+5;
int n,cnt,adj[N],nxt[N<<1],to[N<<1],dep[N],son[N];
inline void addedge(int u,int v){
	nxt[++cnt]=adj[u];adj[u]=cnt,to[cnt]=v;
}
void dfs(int u,int fa){
	for(int e=adj[u];e;e=nxt[e]){
		int v=to[e];
		if(v==fa)continue;
		dfs(v,u);
		if(dep[v]>dep[son[u]])son[u]=v;
	}
	dep[u]=dep[son[u]]+1;
}
int *f[N],tmp[N],*g=tmp,ans[N];
inline void dp(int u,int fa){
	f[u][0]=1;
	if(son[u]){f[son[u]]=f[u]+1,dp(son[u],u),ans[u]=ans[son[u]]+1;}
	for(int e=adj[u];e;e=nxt[e]){
		int v=to[e];
		if(v==fa||v==son[u])continue;
		f[v]=g,g+=dep[v];dp(v,u);
		for(int i=1;i<=dep[v];i++){
			f[u][i]+=f[v][i-1];
			if((f[u][i]>f[u][ans[u]])||(f[u][i]==f[u][ans[u]]&&i<ans[u]))ans[u]=i;
		}
	}
	if(f[u][ans[u]]==1)ans[u]=0;
}
int main(){
	n=read();
	for(int i=1;i<n;i++){
		int u=read(),v=read();
		addedge(u,v),addedge(v,u);
	}
	dfs(1,0);
	f[1]=g,g+=dep[1];
	dp(1,0);
	for(int i=1;i<=n;i++)cout<<ans[i]<<"
";
}