CSP201412-4(kruskal裸题)

最新推荐文章于 2023-03-12 14:35:53 发布

原创最新推荐文章于 2023-03-12 14:35:53 发布 · 252 阅读

0 ·

CC 4.0 BY-SA版权

CSP 同时被 2 个专栏收录

4 篇文章

订阅专栏

KRUSKAL

1 篇文章

订阅专栏

该博客详细介绍了如何解决CSP201412-4问题，强调这是一道典型的最小生成树（Minimum Spanning Tree, MST）题目，可以使用Prim或Kruskal算法。博主选择了Kruskal算法，并以并查集判断麦田数目是否等于总数来结束循环。" 109192125,9727859,重排链表：找到中间结点并翻转后续部分,"['链表操作', '数据结构', '算法题解', 'LeetCode', '递归']

在这里插入图片描述
完全就是一道不带拐弯的minimum spanning tree，prim或kruskal都可以，这里我选择了kruskal。
以并查集1中的麦田数目是否等于麦田总数作为循环跳出条件。

#include <iostream>//考虑用set，但是需要重载运算符，不如用vector算了
#include <vector>
#include <algorithm>
using namespace std;
struct ROAD{
	int vt1,vt2;
	int cost;
};
bool cmp(ROAD a,ROAD b){
	return a.cost<b.cost;
} 
int order=1; 
vector<int> flag(1001);
vector<ROAD> r;
int main(){
	int n,m;
	cin>>n>>m;
	while (m--){
		ROAD d;
		int v1,v2,c;
		scanf ("%d%d%d",&v1,&v2,&c);
		d.vt1=v1;
		d.vt2=v2;
		d.cost=c;
		r.push_back(d);
	}
	sort(r.begin(),r.end(),cmp);
	long long ttcost=0;
	int f=0;
	auto pos=r.begin();
	while (f<n){
		if (!flag[pos->vt1]&&!flag[pos->vt2]){
			ttcost+=pos->cost;
			if (order==1) f+=2;
			flag[pos->vt1]=flag[pos->vt2]=order;
			order++;
		}
		else if (!flag[pos->vt1]){
			ttcost+=pos->cost;
			flag[pos->vt1]=flag[pos->vt2];
			if (flag[pos->vt2]==1) f++;
		}
		else if (!flag[pos->vt2]){
			ttcost+=pos->cost;
			flag[pos->vt2]=flag[pos->vt1];
			if (flag[pos->vt2]==1) f++;
		}
		else if(flag[pos->vt1]!=flag[pos->vt2]){
			ttcost+=pos->cost;
			bool hasone=false;
			int change;
			int needchange;
			if (flag[pos->vt1]==1||flag[pos->vt2]==1){
				change=1;
				flag[pos->vt1]==1 ? needchange=flag[pos->vt2] : needchange=flag[pos->vt1];
				hasone=true;
			}
			else {
				change=flag[pos->vt1];
				needchange=flag[pos->vt2];
			}
			if (hasone){
				for (int i=1;i<=n;i++){
					if (flag[i]==needchange) {
						flag[i]=change;
						f++;
					}
				}
			}
			else
				for (int i=1;i<=n;i++){
					if (flag[i]==needchange) {
						flag[i]=change;
					}
				}
		}
		pos++;
	}
	printf ("%lld",ttcost);
}