codeforces 1009F（长链刨分加二维数组优化）

最新推荐文章于 2022-09-19 14:54:52 发布

不要默认的人生

最新推荐文章于 2022-09-19 14:54:52 发布

阅读量224

点赞数

分类专栏：长链刨分

原文链接：https://www.cnblogs.com/HocRiser/p/10416144.html

版权

长链刨分专栏收录该内容

2 篇文章 0 订阅

订阅专栏

思路：长链刨分，找出最长链，进行从下到上的dp，哪层多就选哪个。

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
int head[N],nxt[N<<1],to[N<<1];
int son[N],dep[N];
int cnt;
int tmp[N],*f[N],*g=tmp;
int ans[N];
void add(int u,int v)
{
    to[++cnt]=v;
    nxt[cnt]=head[u];
    head[u]=cnt;
}
//长链刨分求节点信息
void dfs(int u,int fa)
{
    for(int i=head[u];i;i=nxt[i])
    {
        if(to[i]!=fa)
        {
            dfs(to[i],u);
            if(dep[to[i]]>dep[son[u]])
                son[u]=to[i];
        }
    }
    dep[u]=dep[son[u]]+1;
}
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 i=head[u];i;i=nxt[i])
    {
        int v=to[i];

        if(v==fa||v==son[u])continue;printf("%d\n",v);
            f[v] = g;g+=dep[v];dp(v,u);
            for(int j=1;j<=dep[v];j++)
            {
                f[u][j]+=f[v][j-1];
                if(f[u][j]>f[u][ans[u]] || (f[u][j]==f[u][ans[u] ] && j<ans[u]))
                    ans[u]=j;
            }

    }
    if(f[u][ans[u] ]==1)ans[u] =0;//如果这个点的层里是一，改为一
}
int n;
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n-1;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);add(b,a);
    }
    dfs(1,0);
    f[1]=g;g+=dep[1];
    dp(1,0);
    for(int i=1;i<=n;i++)
        printf("%d\n",ans[i]);
   // for(int i=0;i<5;i++)
     //   printf("%d\n",f[1][i]);
       // printf("%d",dep[2]);
}