总结20220215（kruskal和prim）

今天看了一下最小生成树，原理看懂了，并且会用了kruskal算法和prim算法的模板。下午写了几个kruskal算法的模板题。

最小生成树

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 orz。

输入格式

第一行包含两个整数 N,MN,M，表示该图共有 NN 个结点和 MM 条无向边。

接下来 MM 行每行包含三个整数 X_i,Y_i,Z_iXi​,Yi​,Zi​，表示有一条长度为 Z_iZi​ 的无向边连接结点 X_i,Y_iXi​,Yi​。

输出格式

如果该图连通，则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz。

输入输出样例

输入 #1复制

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

输出 #1复制

7

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 10; 
struct node {
	int x,y,z;
}edge[maxn];
bool cmp(node a,node b) {
	return a.z < b.z;
}
int fa[maxn];
int n,m;
int u,v,w; 
long long sum;
int get(int x) {
	return x == fa[x] ? x : fa[x] = get(fa[x]);
}

int main(void) {
	scanf("%d%d",&n,&m);
	for(int i = 1; i <= m; i ++) {
		scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].z);
	}
	for(int i = 0; i <= n; i ++) {
		fa[i] = i;
	}
	sort(edge + 1,edge + 1 + m,cmp);
	// 每次加入一条最短的边
	for(int i = 1; i <= m; i ++) {
		int x = get(edge[i].x);
		int y = get(edge[i].y);
		if(x == y) continue;
		fa[y] = x;
		sum += edge[i].z;
	}
	int ans = 0;
	for(int i = 1; i <= n; i ++) {
		if(i == fa[i]) ans ++;
	}
	if(ans > 1) puts("orz");
	else printf("%lld",sum);
	return 0;
} 

拆地毯

题目背景

还记得 NOIP 2011 提高组 Day1 中的铺地毯吗？时光飞逝，光阴荏苒，三年过去了。组织者精心准备的颁奖典礼早已结束，留下的则是被人们踩过的地毯。请你来解决类似于铺地毯的另一个问题。

题目描述

会场上有 n 个关键区域，不同的关键区域由 m 条无向地毯彼此连接。每条地毯可由三个整数 u、v、w 表示，其中 u 和 v 为地毯连接的两个关键区域编号，w 为这条地毯的美丽度。

由于颁奖典礼已经结束，铺过的地毯不得不拆除。为了贯彻勤俭节约的原则，组织者被要求只能保留 K 条地毯，且保留的地毯构成的图中，任意可互相到达的两点间只能有一种方式互相到达。换言之，组织者要求新图中不能有环。现在组织者求助你，想请你帮忙算出这 K 条地毯的美丽度之和最大为多少。

输入格式

第一行包含三个正整数 n、m、K。

接下来 m 行中每行包含三个正整数 u、v、w。

输出格式

只包含一个正整数，表示这 K 条地毯的美丽度之和的最大值。

输入输出样例

输入 #1复制

5 4 3
1 2 10
1 3 9
2 3 7
4 5 3

输出 #1复制

22

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 10; 
struct node {
	int x,y,z;
}edge[maxn];
bool cmp(node a,node b) {
	return a.z > b.z;
}
int fa[maxn];
int n,m,p;
int u,v,w; 
long long sum;
int get(int x) {
	return x == fa[x] ? x : fa[x] = get(fa[x]);
}

int main(void) {
	scanf("%d%d%d",&n,&m,&p);
	for(int i = 1; i <= m; i ++) {
		scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].z);
	}
	for(int i = 0; i <= n; i ++) {
		fa[i] = i;
	}
	sort(edge + 1,edge + 1 + m,cmp);
	// 每次加入一条最短的边
	int ans = 0;
	for(int i = 1; i <= m; i ++) {
		int x = get(edge[i].x);
		int y = get(edge[i].y);
		if(x == y) continue;
		fa[y] = x;
		sum += edge[i].z;
		ans++;
		if(ans==p)break;
	}
	printf("%lld",sum);
	return 0;
}