P1195 口袋的天空（kruskal最小生成树）（内附封面）

口袋的天空

题目背景

小杉坐在教室里，透过口袋一样的窗户看口袋一样的天空。

有很多云飘在那里，看起来很漂亮，小杉想摘下那样美的几朵云，做成棉花糖。

题目描述

给你云朵的个数 $N$，再给你 $M$ 个关系，表示哪些云朵可以连在一起。

现在小杉要把所有云朵连成 $K$ 个棉花糖，一个棉花糖最少要用掉一朵云，小杉想知道他怎么连，花费的代价最小。

输入格式

第一行有三个数 $N,M,K$。

接下来 $M$ 行每行三个数 $X,Y,L$，表示 $X$ 云和 $Y$ 云可以通过 $L$ 的代价连在一起。

输出格式

对每组数据输出一行，仅有一个整数，表示最小的代价。

如果怎么连都连不出 $K$ 个棉花糖，请输出 No Answer。

样例 #1

样例输入 #1

3 1 2
1 2 1

样例输出 #1

1

提示

对于 $30\%$ 的数据，$1\le N\le 100$，$1\le M\le 10^3$；

对于 $100\%$ 的数据，$1\le N\le 10^3$，$1\le M\le 10^4$，$1\le K\le 10$，$1\le X,Y\le N$，$0\le L<10^4$。

本题是非常简单的最小生成树的问题~~

最小生成树

最小生成树有两种算法，分别为 kruskal 和 prim ，两者的原理都是贪心

$$kruskal实现:$$

• 将所有边从小到大排序；
• 将所有点分别放入各自的并查集中
• 选择所有未使用边中边权最小的
• 若该边所连接的两个点已经联通，则舍去，否则合并这两个并查集
• 重复上述两步，直到所有点都已经被标记或所有未标记的点都不存在连边与已标记的点直接相连

并查集是判断两个点是否已经属于一个连通块的工具

int find(int x){
   if(fa[x]==x)return x;
   return fa[x]=find(fa[x]);//路径压缩
}
void merge(int x,int y){
   fa[find(x)]=find(y);
}

kruskal部分（对于本题，要求连接k个边，因此已连接k个边后break掉）

void kruskal(){
	int cnt=0;
	sort(a+1,a+1+m,cmp);
	for(int i=1;i<=n;i++){
		fa[i]=i;
	}
	for(int i=1;i<=m;i++){
		if(n-k<=cnt)break;
		if(find(a[i].u)!=find(a[i].v)){
			cnt++;
			ans+=a[i].w;
			merge(a[i].u,a[i].v);
		}
	}
	
} 

$$prim实现$$

• 维护一个连通块，每次向外延伸一条最短边，将新的点合并进连通块内。
• 用一个数组记录当前点是否被选中连接。
• 将所有可能延伸到的点入队（小根堆，优先队列优化）

#include <bits/stdc++.h>
#define N 5005
#define M 200005
using namespace std;

typedef pair<int, int> pii;

int dis[N];
int n, m;
bool vis[N];
vector<pii> edges[N];
priority_queue<pii, vector<pii>, greater<pii> > Q;
int main() {
    cin >> n >> m;
    for (int i = 0, u, v, w; i < m; i++) {
        cin >> u >> v >> w;
        edges[u].emplace_back(v, w);
        edges[v].emplace_back(u, w);
    }
    Q.push({0, 1});
    int reachedPoints = 0;
    int totWeight = 0;
    memset(dis, 0x3f, sizeof(dis));
    dis[1] = 0;
    while (!Q.empty() && reachedPoints < n) {
        pii info = Q.top();
        Q.pop();
        int x = info.second;
        if (vis[x])
            continue;
        vis[x] = true;
        reachedPoints++;
        totWeight += info.first;
        for (auto edge: edges[x]) {
            if (!vis[edge.first] && dis[edge.first] > edge.second) {
                dis[edge.first] = edge.second;
                Q.push({dis[edge.first], edge.first});
            }
        }
    }
    return 0;
}

AC CODE

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+23;
int n,m,k,fa[N];
int ans=0;
struct node{
	int u,v,w;
}a[N];
bool cmp(node aa,node bb){
	return aa.w<bb.w;
}
int find(int x){
	if(fa[x]==x)return x;
	return fa[x]=find(fa[x]);
}
void merge(int x,int y){
	fa[find(x)]=find(y);
}
void kruskal(){
	int cnt=0;
	sort(a+1,a+1+m,cmp);
	for(int i=1;i<=n;i++){
		fa[i]=i;
	}
	for(int i=1;i<=m;i++){
		if(n-k<=cnt)break;
		if(find(a[i].u)!=find(a[i].v)){
			cnt++;
			ans+=a[i].w;
			merge(a[i].u,a[i].v);
		}
	}
	
} 
int main(){
	cin>>n>>m>>k;
	for(int i=1;i<=m;i++){
		cin>>a[i].u>>a[i].v>>a[i].w;
	}
	kruskal();
	cout<<ans<<endl;
	return 0;
}

附封面（天气之子）