[CF1009F][长链剖分]Dominant Indices

CF1009F

长链剖分模板题
定义f[i][j]为子树i中到i的距离为j的点的个数
然后就是套路了

长链剖分简单讲解：
按照dep划分重儿子（称作长儿子有点奇怪）
然后合并信息的时候先直接继承重儿子信息，然后暴力合并轻儿子信息
因为一个点只会在一条长链中，一条长链只会被合并一次，所以是$O(n)$的

空间开不下？
动态分配内存

Code：

#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<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=1e6+5;
int vis[N<<1],head[N<<1],nxt[N<<1],tot=0;
inline void add(int x,int y){vis[++tot]=y;nxt[tot]=head[x];head[x]=tot;}
int pt[N],len[N],lson[N];
int n;
void dfs(int v){
	pt[v]=1;
	for(int i=head[v];i;i=nxt[i]){
		if(pt[vis[i]]) continue;
		dfs(vis[i]);
		if(len[vis[i]]>len[lson[v]]) lson[v]=vis[i];
	}
	len[v]=len[lson[v]]+1;
}
int *f[N],tmp[N],*cnt=tmp,ans[N];//动态分配内存 
void dp(int v){
	f[v][0]=1;pt[v]=1;
	if(lson[v]) {f[lson[v]]=f[v]+1;dp(lson[v]);ans[v]=ans[lson[v]]+1;}
	for(int i=head[v];i;i=nxt[i]){
		if(vis[i]==lson[v] || pt[vis[i]]) continue;
		f[vis[i]]=cnt,cnt+=len[vis[i]];
		dp(vis[i]);
		for(int j=1;j<=len[vis[i]];j++){
			f[v][j]+=f[vis[i]][j-1];
			if((j<ans[v] && f[v][j]>=f[v][ans[v]]) || (j>ans[v] && f[v][j]>f[v][ans[v]])) ans[v]=j;
		}
	}
	if(f[v][ans[v]]==1) ans[v]=0;
}
int main(){
	n=read();
    for(int x,y,i=1;i<n;i++) x=read(),y=read(),add(x,y),add(y,x);
    dfs(1);memset(pt,0,sizeof(pt));
	f[1]=cnt;cnt+=len[1];
    dp(1);
    for(int i=1;i<=n;++i) cout<<ans[i]<<"\n";
    return 0;
}