【tyvj P1307】联络员

题解:

        简单应用,先选中必选边,再在可选边中求一次最小生成树。

#include <stdio.h>
#include <iostream>
#include <algorithm>
using namespace std;
int n,m,cnt,ans,father[2000+10];
struct data{
	int u,v,w;
}e[10000+10];
bool cmp(const data &a,const data &b)
{
	return a.w<b.w;
}
int find(int x)
{
	if(x!=father[x]) father[x]=find(father[x]);
		return father[x];
}
void insert(int x,int y,int v)
{
	cnt++;
	e[cnt].u=x;e[cnt].v=y;e[cnt].w=v;
}
void Union(int x,int y)
{
	x=find(x),y=find(y);
	if(x!=y) father[x]=y;
}
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
		father[i]=i;
	int x,y;
	for(int i=1;i<=m;i++)
	{
		int p,u,v,w;
		scanf("%d %d %d %d",&p,&u,&v,&w);
		if(p==1)
		{
			Union(u,v);
			ans+=w;
		}else
		{
			insert(u,v,w);
		}
	}
	sort(e+1,e+cnt+1,cmp);
	for(int i=1;i<=cnt;i++)
	{
		x=find(e[i].u),y=find(e[i].v);
		if(x!=y)
		{
			Union(x,y);
			ans+=e[i].w;
		}
	}
	printf("%d\n",ans);
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值