首页 > 代码库 > Hdu 2196 Computer (树形dp)

Hdu 2196 Computer (树形dp)

题目大意:

求出每一个结点距离叶子节点的最大距离。


思路分析:

假设1为树的根。

通过一遍dfs 可以求出每个节点到儿子节点上的叶子节点的最远距离。

然后我们要做的就是如果更新答案。

问题就是如果我们开始求的最大值本来就是一个儿子更新的,如果再深搜的话又会造成重复更新。

所以就记录两个值,最大值和次大值。

然后不断更新这两个值。


#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#define maxn 10005

using namespace std;

int head[maxn];
int to[maxn<<1];
int next[maxn<<1];
int len[maxn<<1];
int tot;

void init()
{
    tot=0;
    memset(head,0,sizeof head);
}

void addedge(int u,int v,int w)
{
    tot++;
    next[tot]=head[u];
    to[tot]=v;
    head[u]=tot;
    len[tot]=w;
}

bool vis[maxn];
int maxval[maxn],nex[maxn],maxid[maxn],nexid[maxn];
//最大值        次大值   更新最大值的id  更新次大值的id
void dfs(int x)
{
    maxval[x]=nex[x]=0;
    for(int p = head[x] ;p; p = next[p])
    {
        int v = to[p];
        if(!vis[v])
        {
            vis[v]=true;
            dfs(v);
            if(nex[x]<maxval[v]+len[p])
            {
                nex[x]=maxval[v]+len[p];
                nexid[x]=v;
                if(nex[x]>maxval[x])
                {
                    swap(nex[x],maxval[x]);
                    swap(nexid[x],maxid[x]);
                }
            }
        }
    }
}

void update(int x,int pre)
{
    for(int p = head[x];p;p=next[p])
    {
        int v=to[p];
        if(v==pre)continue;

        if(v==maxid[x])
        {
            if(nex[x]+len[p]>nex[v])
            {
                nex[v]=nex[x]+len[p];
                nexid[v]=x;

                if(nex[v]>maxval[v])
                {
                    swap(nex[v],maxval[v]);
                    swap(nexid[v],maxid[v]);
                }
            }
        }
        else {
            if(maxval[x]+len[p]>nex[v])
            {
                nex[v]=maxval[x]+len[p];
                nexid[v]=x;

                if(nex[v]>maxval[v])
                {
                    swap(nex[v],maxval[v]);
                    swap(nexid[v],maxid[v]);
                }
            }
        }
        update(v,x);
    }
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        init();

        for(int i=2;i<=n;i++)
        {
            int l,r;
            scanf("%d%d",&l,&r);
            addedge(i,l,r);
            addedge(l,i,r);
        }

        memset(vis,false,sizeof vis);
        vis[1]=true;
        dfs(1);
        update(1,-1);
        for(int i=1;i<=n;i++)
            printf("%d\n",maxval[i]);
    }
    return 0;
}