CF1009F Dominant Indices
作者:互联网
I.III.CF1009F Dominant Indices
大约一年前,我曾用线段树合并硬刚过了这题,现在是正解。
可以设 \(f_{i,j}\) 表示 \(i\) 子树中深度为 \(j\) 的点的数量。于是直接长剖优化即可。时间复杂度 \(O(n)\)。
虽然因为我极其垃圾的实现导致还没有线段树合并跑得快
代码:
#include<bits/stdc++.h>
using namespace std;
int n,*f[1001000],F[1001000],dep[1001000],son[1001000],g[1001000],*ff=F;
vector<int>v[1001000];
void dfs1(int x,int fa){
for(auto y:v[x])if(y!=fa){
dfs1(y,x);
if(dep[y]>dep[son[x]])son[x]=y;
}
dep[x]=dep[son[x]]+1;
}
void dfs2(int x,int fa){
if(!f[x])f[x]=ff,ff+=dep[x];
if(son[x])f[son[x]]=f[x]+1,dfs2(son[x],x),g[x]=g[son[x]]+1;
f[x][0]=1;if(f[x][g[x]]<=1)g[x]=0;
for(auto y:v[x])if(y!=fa&&y!=son[x]){
dfs2(y,x);
for(int i=0;i<dep[y];i++){
f[x][i+1]+=f[y][i];
if(f[x][i+1]>f[x][g[x]]||f[x][i+1]==f[x][g[x]]&&g[x]>i+1)g[x]=i+1;
}
}
}
int main(){
scanf("%d",&n);
for(int i=1,x,y;i<n;i++)scanf("%d%d",&x,&y),v[x].push_back(y),v[y].push_back(x);
dfs1(1,0),dfs2(1,0);
// for(int i=1;i<=n;i++)printf("(%d %d)\n",dep[i],son[i]);
for(int i=1;i<=n;i++)printf("%d\n",g[i]);
return 0;
}
标签:Dominant,CF1009F,fa,int,1001000,son,dep,ff,Indices 来源: https://www.cnblogs.com/Troverld/p/14636997.html