hdu 2196 Computer (树形dp+两次dfs遍历)

来源：互联网发布：轻松玩转scratch编程编辑：程序博客网时间：2024/06/06 02:22

题意：n个点链接成树，已知2-n个点与点xi相邻，连接他们的边长为yi(xi、yi分别是

第2行到第n行的两个数）。求树上离每个点的最远距离。

分析：暂略

#include<cstdio>#include<iostream>#include<cstring>#include<vector>using namespace std;vector<int> son[10010],w[10010];int f[10010],g[10010],dp[10010],longest[10010];int maxx(int a,int b){    return a>b?a:b;}int dfs1(int u)   //u为根节点{    if(f[u]) return f[u];    int s=son[u].size();    if(!s) return 0;    //叶子节点    int eri=-1,er=-1,esti,est=-1;    for(int i=0;i<s;i++)    {        int v=son[u][i];        if(dfs1(v)+w[u][i]>est)        {            est=f[v]+w[u][i];            esti=i;        }    }    longest[u]=esti;    f[u]=est;    for(int i=0;i<s;i++)    {        int v=son[u][i];        if(dfs1(v)+w[u][i]>er && i!=esti)        {            er=f[v]+w[u][i];            eri=i;        }    }    if(eri!=-1) g[u]=er;    return f[u];}void dfs2(int u){    int s=son[u].size();    for(int i=0;i<s;i++)    {        int t=son[u][i];        if(longest[u]==i)            dp[t]=maxx(g[u],dp[u])+w[u][i];        else            dp[t]=maxx(f[u],dp[u])+w[u][i];        dfs2(t);    }}int main(){    int n,a,b;    while(scanf("%d",&n)!=EOF)    {        memset(longest,-1,sizeof(longest));        memset(dp,0,sizeof(dp));        memset(f,0,sizeof(f));        memset(g,0,sizeof(g));        for(int i=0;i<n;i++)        {            son[i].clear();            w[i].clear();        }        for(int i=2;i<=n;i++)        {            scanf("%d%d",&a,&b);            son[a].push_back(i);            w[a].push_back(b);        }        f[1]=dfs1(1);        dp[1]=0;        dfs2(1);        for(int i=1;i<=n;i++)            printf("%d\n",maxx(dp[i],maxx(f[i],g[i])));    }    return 0;}//f[]是当前节点往子树方向所能走过的最远距离//g[]是往子树方向次远距离//dp[]是通过父节点能走过的最远距离//longest[]用来记录当前节点的最远路径所对应的分支编号//（因为是邻接表存储的，对于当前节点的判重只要记录下分支编号就行了）

0 0