hdu 4313 Matrix

最新推荐文章于 2024-07-31 10:00:00 发布

菜鸟起航

最新推荐文章于 2024-07-31 10:00:00 发布

阅读量571

点赞数

CC 4.0 BY-SA版权

分类专栏：并差集

本文链接：https://blog.youkuaiyun.com/aixiaoling1314/article/details/8794978

并差集专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种使用并查集处理特定类型问题的方法。通过将边按权重从大到小排序，并在处理过程中确保合并操作使得机器节点成为根节点的方式，实现对图中连接的有效管理和维护。

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

并查集
初始时没有边加入，将所有边按权值从大到小排序，然后依次判断每一个边两端的顶点是否是均为machine节点，

如果是则应删除这条边，否则加入这条边，然后在并查集合并时尽量让根节点为machine节点。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm> 
using namespace std;
#define N 100100
struct edge
{
	int s,d;
	int cost;
}e[N];
int num=0,n,f[N];
bool vis[N];
int find(int a)
{
	if(a!=f[a])
		f[a]=find(f[a]);
	return f[a];
}
bool cmp(struct edge a,struct edge b)
{
	return a.cost>b.cost;
}	
void addedge(int a,int b,int c)
{
	e[num].s=a;e[num].d=b;
	e[num++].cost=c;
}
int main()
{
	int i,j,k,t,a,b,c;
	__int64 sum;
	scanf("%d",&t);
	while(t--)
	{
		num=0;
		scanf("%d%d",&n,&k);
		memset(vis,false,sizeof(vis));
		for(i=0;i<n;i++)
			f[i]=i;
		for(i=1;i<n;i++)
		{
			scanf("%d%d%d",&a,&b,&c);
			addedge(a,b,c);
		}
		for(i=0;i<k;i++)
		{
			scanf("%d",&a);
			vis[a]=true;
		}
		sort(e,e+num,cmp);
		sum=0;
		for(i=0;i<num;i++)
		{
			a=find(e[i].s);b=find(e[i].d);
			if(a!=b&&vis[a]==true&&vis[b]==true)
			{
				sum+=e[i].cost;
			}
			else 
			{
				if(vis[a])
					f[b]=a;
				else
					f[a]=b;	
			}
		}
		printf("%I64d\n",sum);
	}
	return 0;
}