HDU-4313-Matrix

最新推荐文章于 2024-06-28 02:35:22 发布

南宮逸辰

最新推荐文章于 2024-06-28 02:35:22 发布

阅读量407

点赞数

CC 4.0 BY-SA版权

分类专栏： HDU ACM 文章标签：最小生成树

本文链接：https://blog.youkuaiyun.com/z309241990/article/details/37660497

ACM 同时被 2 个专栏收录

716 篇文章

订阅专栏

HDU

131 篇文章

订阅专栏

本文介绍一种基于最小生成树的反向求解算法，通过从大到小排序边的权重，实现特定节点间边权值之和的计算。适用于解决特定图论问题。

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

这个题意比较简单，做法就是最小生成树的反向求解（排序从大到小)，只要当前并查集查出2个都为要求点，那么这条边的权值就应该加上。否则合并

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<string>
#include<algorithm>
using namespace std;
const int inf=1<<29;
const int maxn=1e5+1000;
struct Node
{
    int u;
    int v;
    int c;
    bool operator < (const Node &a)const
    {
        return c>a.c;
    }
}a[maxn];
int n,m,p[maxn];
bool is[maxn];
int find(int x)
{
    if(p[x]==-1)
        return x;
    return p[x]=find(p[x]);
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(is,0,sizeof(is));
        memset(p,-1,sizeof(p));
        scanf("%d%d",&n,&m);
        for(int i=0;i<n-1;i++)
            scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].c);
        sort(a,a+n-1);
        for(int i=0;i<m;i++)
        {
            int u;
            scanf("%d",&u);
            is[u]=1;
        }
        long long ans=0;
        for(int i=0;i<n-1;i++)
        {
            int x=find(a[i].u);
            int y=find(a[i].v);
            if(is[x]&&is[y])
            {
                ans+=a[i].c;
            }
            else if(is[x])
            {
                p[y]=x;
            }
            else
            {
                p[x]=y;
            }
        }
        printf("%I64d\n",ans);

    }
    return 0;
}