HDU-2196 树的直径性质

HDU-2196
本题大意是求取一颗无根树上每一个节点,到树上其余任意节点的最远距离

其实我们可以考虑利用树的直径的性质,我们假设树的直径的两个端点为 AB那么,树上任意节点C到树上其余任何节点的最远距离一定是C->A或者是C->B
利用这个性质,我们只需利用3次dfs,
第一次是为了求取直径的A端点
第二次是为了求取其余所有点到A端点的距离,顺便可以求取B端点
最后我们可以再求取一次所有点到B端点的距离

对于每一个节点我们可以记录一个 ans[i]=max(ans[i],sum[i])
这样就可以保证求出答案的正确性
AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <queue>
#include <functional>
#include <vector>
#include <stack>
#include <set>
#include <string>
#define ls k*2
#define rs k*2+1
#define int long long
using namespace std;
typedef long long ll;
const int maxn=1e6+50;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
int HASH=131;


struct node
{
    int to,val;
    node(int a,int b):to(a),val(b){};
};

vector<node> v[maxn];
int ans[maxn];
int sum[maxn];
int maxx=0;
int id=0;

void dfs(int u,int father)
{
    if(maxx<sum[u])
    {
        maxx=sum[u];
        id=u;
    }
    for(int i=0;i<v[u].size();i++)
    {
        node tmp=v[u][i];
        if(tmp.to==father) continue;
        sum[tmp.to]=sum[u]+tmp.val;
        ans[tmp.to]=max(sum[tmp.to],ans[tmp.to]);
        dfs(tmp.to,u);
    }
    return;
}


signed main()
{
    int n;
    while(scanf("%lld",&n)!=EOF)
    {
        for(int i=1;i<=n;i++)
            v[i].clear();
        for(int i=2;i<=n;i++)
        {
            int x,val;
            scanf("%lld %lld",&x,&val);
            v[i].push_back(node(x,val));
            v[x].push_back(node(i,val));
        }
        for(int i=1;i<=n;i++)
        {
            sum[i]=0;
            ans[i]=0;
        }
        maxx=0;
        dfs(1,1);
        for(int i=1;i<=n;i++)
            sum[i]=0;
        maxx=0;
        dfs(id,id);
        for(int i=1;i<=n;i++)
            sum[i]=0;
        maxx=0;
        dfs(id,id);
        for(int i=1;i<=n;i++)
        {
            printf("%lld\n",ans[i]);
        }
    }
    return 0;
}


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值