hdu 2196 树形dp

题意:   求每个点的最远距离。

树形dp,做了两天,感觉就是从子节点得到父节点。


此题要用到两次dfs, 第一次dfs1用来求所有节点在他子树范围内到叶子节点的最长距离和次长距离。dp1,和dp2

第二次dfs2 ,求f【】,如果dp1[s]==dp1[x]+len (s为父节点,x为子节点),则 x 在最长树的分支上,f[x]=max(f[s]+len,dp2[s]+len); 

所以f【x】就等于f父节点加上到子节点的距离和父节点的次长加上到x节点的距离的两者的最大值。

如果 不相等,则代表x节点不在父节点最长的分支上。则f【x】就等于f父节点到x节点的距离和父节点的最长距离加上到子节点距离 两者的最大值。

结果在max(f【】,dp1【】);

#include<cstdio>
#include<cstring>
#include<vector>
#include<iostream>
using namespace std;
const int Max=10001;
struct Point
{
    int v,w;
};
vector<Point>V[Max];
int vis[Max];
int f[Max],dp1[Max],dp2[Max];
void init()
{
    for(int i=1;i<Max;i++)
    {
        V[i].clear();
    }
    memset(dp1,0,sizeof(dp1));
    memset(dp2,0,sizeof(dp2));
    memset(vis,0,sizeof(vis));
    memset(f,0,sizeof(f));
}
void dfs1(int s) //所有节点在他子树范围内到叶子节点的最长距离和次长距离。dp1,和dp2
{
    vis[s]=1;
    for(int i=0;i<(int)V[s].size();i++)
    {
        int x=V[s][i].v;
        int len=V[s][i].w;
        if(!vis[x])
        {
            dfs1(x);
            int a=max(max(dp1[s],dp2[s]),dp1[x]+len);
            if(a==dp1[s]) dp2[s]=max(dp2[s],dp1[x]+len);
            else if(a==dp2[s]) dp2[s]=max(dp1[s],dp1[x]+len);
            else dp2[s]=max(dp1[s],dp2[s]);
            dp1[s]=a;
        }
    }
}
void dfs2(int s)  //第二次dfs,求f数组。
{
    vis[s]=1;
    for(int i=0;i<(int)V[s].size();i++)
    {
        int x=V[s][i].v;
        int len=V[s][i].w;
        if(!vis[x])
        {
            if(dp1[s]==dp1[x]+len)
               f[x]=max(f[s]+len,dp2[s]+len);
            else
              f[x]=max(f[s]+len,dp1[s]+len);
            dfs2(x);
        }
    }
}
int main()
{
    //freopen("Input.txt","r",stdin);
    int n,i;
    while(~scanf("%d",&n))
    {
        init();
        int a,b;
        for(i=2;i<=n;i++)
        {
            scanf("%d%d",&a,&b);
            Point tmp;
            tmp.v=a;tmp.w=b;
            V[i].push_back(tmp);
            tmp.v=i;
            V[a].push_back(tmp);
        }
        dfs1(1);
        memset(vis,0,sizeof(vis));
        dfs2(1);
        for(i=1;i<=n;i++)
          printf("%d\n",max(dp1[i],f[i]));
    }
    return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值