[NOI2011]道路修建

最新推荐文章于 2021-05-16 16:06:14 发布

weixin_30950607

最新推荐文章于 2021-05-16 16:06:14 发布

阅读量119

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/bullshit/p/9656412.html

本文深入解析了一道关于树论的算法题目，通过代码实现展示了如何计算树中节点间的距离，并结合具体实例说明了算法的运行过程及注意事项。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

树论

一道大水题，直接放代码：

code：

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<cstring>
#define int long long 
#define half (l+r)>>1
const int maxn=2000006; 
using namespace std;
int head[maxn],cur,d[maxn],hi[maxn];
int qa[maxn],qb[maxn],qc[maxn];
struct hzw
{
    int to,next,v;
}e[maxn];
inline void add(int a,int b,int c)
{
    e[cur].to=b;
    e[cur].next=head[a];
    e[cur].v=c;
    head[a]=cur++;
}
inline void dfs(int s,int fa)
{
    d[s]=1;
    hi[s]=hi[fa]+1;
    for (int i=head[s];i!=-1;i=e[i].next)
    {
        if (e[i].to==fa) continue;
        dfs(e[i].to,s);
        d[s]+=d[e[i].to];
    }
}
inline int abs_(int x,int y)
{
    return x>y?x-y:y-x;
}
int n;
signed main()
{
    memset(head,-1,sizeof(head));
    cin>>n;
    for (int i=1,a,b,c;i<=n-1;++i)
    {
        scanf("%lld%lld%lld",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
        qa[i]=a;
        qb[i]=b;
        qc[i]=c;
    }
    dfs(1,1);
    int ans=0;
    for (int i=1;i<=n-1;++i)
    {
        int x=qa[i],y=qb[i],z=qc[i];
        int tmp=hi[x]>hi[y]?d[x]:d[y];
        ans=ans+z*abs_(tmp,n-tmp);
    }
    cout<<ans;
    return 0;
}