mid-term 1005 snow road

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

struct road{
	int a,b,distanc;
};
int T,n,m1,m2;
int root[100];
int rankk[100];
road grah[1001];

bool cmp(const road &a,const road &b)
{
	return a.distanc<b.distanc;
}

int roott(int a)
{
	if( a==root[a])
		return a;
	else
		return root[a]=roott(root[a]);
}

void merge(int a,int b)
{
	int roota=roott(a);
	int rootb=roott(b);
	if (rankk[roota]>rankk[rootb])
	{
		root[rootb]=roota;
	}
	else if (rankk[roota]<rankk[rootb])
	{
		root[roota]=rootb;
	}
	else
	{
		root[roota]=rootb;
		rankk[rootb]++;
	}
}

int main()
{
	cin>>T;
	while(T--)
	{
		cin>>n>>m1>>m2;
		for (int i=0;i<n;i++)
		{
			root[i]=i;
		}
		int aa=0;
	    while (m1--)
	    {
			int a1,a2;
			cin>>a1>>a2;
			grah[aa].a=a1;
			grah[aa].b=a2;
			grah[aa].distanc=0;
			aa++;
		}
		while (m2--)
		{
			int a1,a2,a3;
			cin>>a1>>a2>>a3;
			grah[aa].a=a1;
			grah[aa].b=a2;
			grah[aa].distanc=a3;
			aa++;
		}
		sort(grah,grah+aa,cmp);
		for (int i=0;i<n;i++)
		{
			rankk[i]=0;
		}
		int sumofroad=0;
		for (int i=0;i<aa;i++)
		{
			int begi=grah[i].a;
			int endd=grah[i].b;
			int weig=grah[i].distanc;
			if(roott(begi)!=roott(endd))
			{
				merge(begi,endd);
				sumofroad+=weig;
			}
		}
		cout<<sumofroad<<endl;
	}
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值