ZOJ3659,2012长春现场赛E题,并查集

ACM_Victoria

于 2012-10-15 21:52:59 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM 文章标签： struct

本文链接：https://blog.youkuaiyun.com/yihuikang/article/details/8074897

ACM 专栏收录该内容

94 篇文章

订阅专栏

本文介绍了一种基于最小生成树算法解决特定问题的方法，并通过C++实现了解决方案。该算法利用了并查集进行节点合并，实现了高效的边权累积。通过对每条边的权值进行比较与合并操作，最终求得了一个顶点集合的最大价值。

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

赛后在麦当劳由科大一牛提点才会的~

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long LL;

const int NN=250000;

struct Edge{
    int u,v;
    LL w;
    bool operator <(const Edge a)const{
        return w>a.w;
    }
} e[NN];

int belong[NN];
struct Set{
    int num;
    LL value;
} a[NN];

int find(int x)
{
    if (belong[x]!=x) belong[x]=find(belong[x]);
    return belong[x];
}

void set_union(int u,int v,LL w)
{
    if (a[u].num<a[v].num)
    {
        belong[u]=v;
        a[v].num+=a[u].num;
        a[v].value=w;
    }
    else
    {
        belong[v]=u;
        a[u].num+=a[v].num;
        a[u].value=w;
    }
}

int main()
{
    int n;
    while (scanf("%d",&n)!=EOF)
    {
        for (int i=1; i<n; i++) scanf("%d%d%lld",&e[i].u,&e[i].v,&e[i].w);
        sort(e+1,e+n);

        for (int i=1; i<=n; i++)
        {
            belong[i]=i;
            a[i].num=1;
            a[i].value=0;
        }

        for (int i=1; i<n; i++)
        {
            int u=find(e[i].u);
            int v=find(e[i].v);
            LL tmp1=a[u].value+a[v].num*e[i].w;
            LL tmp2=a[v].value+a[u].num*e[i].w;
            set_union(u,v,max(tmp1,tmp2));
        }
        
        printf("%lld\n",a[find(1)].value);
    }
    return 0;
}