PIPI OJ 1118: 继续畅通工程(并查集+最小生成树)

最新推荐文章于 2024-06-19 00:27:00 发布

小黑哎呀

最新推荐文章于 2024-06-19 00:27:00 发布

阅读量347

点赞数 1

文章标签： kruskal 快速排序

本文链接：https://blog.csdn.net/Thy__father/article/details/111934853

版权

菜鸟生成记(18)

1118: 继续畅通工程

又双叒叕是最短路径的水题;不同的是,在构造最小生成树前,题目中已经规定一些已经建好了(这些边已经在生成树里面了);从未建好的边中选择最优边加入生成树;直到所有的顶点都被并入一个连通分量(生成树)中;

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef struct st ak;
const int N=1e+4+10;//因为1<n<100,所以1<m<100*(100-1)/2==4950;直接开1万(阔气) 
int pre[N]={0};//并查集数组 
struct st{
	int v1,v2,w;
	st()
	{
		v1=v2=w=0;
	}
}s[N];
int n;
int find(int x)
{//找祖先 
	int t=x;
	while(pre[t]!=t)
	{
		t=pre[t];
	}
	return t;//返回祖先 
}
int cmp(ak a,ak b)
{//sort排序条件 
	return a.w<b.w;
}
int main()
{
	int num=0,sum=0;
	while(cin>>n&&n!=0)
	{
		num=0;
		sum=0;
		for(int i=1;i<=n;i++)
		pre[i]=i;
		
		int m=n*(n-1)/2;
		for(int i=1;i<=m;i++)
		{
			int t1,t2;
			int x,y,z,f;
			cin>>x>>y>>z>>f;
			if(f==1)//1表示该边已经建成了 
			{//直接并查集走起 
				t1=find(x);//找x的祖先 
				t2=find(y);//找y的祖先 
				pre[y]=t1;//y认x的祖先为自己的祖先 
				pre[t2]=t1;//y的祖先认x的祖先为自己的祖先
				//x和x的祖先与y和y的祖先并入一个连通分量(生成树) 
			}
			else//0表示该边还未建成,待选择 
			{// 
				s[num].v1=x,s[num].v2=y,s[num].w=z;
				num++;
			}	
		}
		sort(s,s+num,cmp);//按权值从小到大排序 
		for(int i=0;i<num;i++)//在剩下的边中选择 
		{//并查集 和上面一样,直到所有结点都并入一个连通分量(生成树)中 
			int t1,t2,x,y;
			x=s[i].v1,y=s[i].v2;
			t1=find(x);
			t2=find(y);
			if(t1!=t2)
			{
				pre[y]=t1;
				pre[t2]=t1;
				sum+=s[i].w;//记录生成树的权值 
			}
		}
		cout<<sum<<endl;
	}
	return 0;
}