Hdu 2196 Computer (樹形dp)

標籤：os   io   for   問題   amp   size   ad   ef   

題目大意：

求出每一個結點距離葉子節點的最大距離。

思路分析：

假設1為樹的根。

通過一遍dfs 可以求出每個節點到兒子節點上的葉子節點的最遠距離。

然後我們要做的就是如果更新答案。

問題就是如果我們開始求的最大值本來就是一個兒子更新的，如果再深搜的話又會造成重複更新。

所以就記錄兩個值，最大值和次大值。

然後不斷更新這兩個值。

#include <cstdio>#include <iostream>#include <cstring>#include <algorithm>#define maxn 10005using 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  更新次大值的idvoid 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;}