HDU 2196

47 篇文章 0 订阅

题意:

求树上每个点能走出的最远距离


思路:

树形DP

第一次dfs求出u点向子树方向走的最长路(dis1)和次长路(dis2)

状态转移  tmp[u] = edge + dis1[v]

第二次dfs求出u点向父亲方向走的最长路

状态转移  tmp[u] = edge + ( u == fa1[f] ) ? dis2[f] : dis1[f]


代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define M 10010

int n,tot;
int head[M],fa1[M],fa2[M];
__int64 dis1[M],dis2[M];
struct edge
{
    int v,w,next;
}ed[M*2];

void dfs1(int u,int f)
{
    int i,v;
    __int64 tmp;
    for(i=head[u];~i;i=ed[i].next)
    {
        v=ed[i].v;
        if(v!=f)
        {
            dfs1(v,u);
            tmp=dis1[v]+ed[i].w;
            if(tmp>dis2[u])
            {
                dis2[u]=tmp;
                fa2[u]=v;
            }
            if(dis2[u]>dis1[u])
            {
                swap(dis2[u],dis1[u]);
                swap(fa2[u],fa1[u]);
            }
        }
    }
}

void dfs2(int u,int f)
{
    int i,v;
    __int64 tmp;
    for(i=head[u];~i;i=ed[i].next)
    {
        v=ed[i].v;
        if(v==f)
        {
            tmp=ed[i].w;
            if(fa1[f]==u) tmp+=dis2[f];
            else tmp+=dis1[f];
            if(tmp>dis2[u])
            {
                dis2[u]=tmp;
                fa2[u]=f;
            }
            if(dis2[u]>dis1[u])
            {
                swap(dis2[u],dis1[u]);
                swap(fa2[u],fa1[u]);
            }
            break;
        }
    }
    for(i=head[u];~i;i=ed[i].next)
    {
        v=ed[i].v;
        if(v!=f) dfs2(v,u);
    }
}

int main()
{
    int i,v,w;
    while(~scanf("%d",&n))
    {
        memset(head,-1,sizeof(head));
        tot=0;
        memset(dis1,0,sizeof(dis1));
        memset(dis2,0,sizeof(dis2));
        for(i=2;i<=n;i++)
        {
            scanf("%d%d",&v,&w);
            ed[tot].v=v; ed[tot].w=w; ed[tot].next=head[i]; head[i]=tot++;
            ed[tot].v=i; ed[tot].w=w; ed[tot].next=head[v]; head[v]=tot++;
        }
        dfs1(1,-1);
        dfs2(1,-1);
        for(i=1;i<=n;i++) printf("%I64d\n",max(dis1[i],dis2[i]));
    }
    return 0;
}


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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值