POJ-2421 Constructing Roads

题目链接:https://vjudge.net/problem/POJ-2421

最小生成树问题,先给出图的邻接矩阵,再给出已经两两相连的点,求最小生成树的花费

把下三角矩阵的边加入边集,已经相连的点用并查集标记,再用Kruskal做就可以了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXN=100+10;//点数
const int MAXM=10000+10;//边数
int UF[MAXN];
struct Edge
{
	int u,v,w;
}edge[MAXM];
int tol; //初始化为0 
void addedge(int u,int v,int w)
{
	edge[tol].u=u;
	edge[tol].v=v;
	edge[tol++].w=w;
} 
bool cmp(Edge a,Edge b) 
{
	return a.w<b.w; 
}
int find(int x) 
{  
	if(UF[x]==-1) return x;  
	else return UF[x]=find(UF[x]);
}
int cnt;
int Kruskal(int n)
{
	sort(edge,edge+tol,cmp);
	int ans=0;
	for(int i=0;i<tol;i++)
	{
		int u=edge[i].u;   
		int v=edge[i].v;   
		int w=edge[i].w; 
		int t1=find(u);
		int t2=find(v);
		if(t1!=t2)
		{
			UF[t1]=t2;
			ans+=w;
			cnt++;
		}
		if(cnt==n-1) break;
	}
	if(cnt<n-1) return -1;
	return ans;
}
int main()
{
	int n,m,v,a,b;
	while(~scanf("%d",&n))
	{
		tol=cnt=0;
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				scanf("%d",&v);
				if(j<i)
				addedge(i,j,v);
			}
		}
		memset(UF,-1,sizeof(UF));
		scanf("%d",&m);
		for(int i=0;i<m;i++)
		{
			scanf("%d%d",&a,&b);
			int t1=find(a);
			int t2=find(b);
			if(t1!=t2) 
			{
				UF[t1]=t2;
				cnt++;
			}
		}
		printf("%d\n",Kruskal(n));
	}
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值