【洛谷】P1195 口袋的天空 的题解

【洛谷】P1195 口袋的天空 的题解

题目传送门

题解

这和板子有什么区别

这题其实也是一道最小生成树的板子题，就是把板子改一改就可以的。

首先，把边从小到大排个序，如果没有回路，一条一条往里加，边数达到 $n-1$ 条。

让我们选 $k$ 朵云做棉花糖，换句话说就是选 $k$ 个节点构造最小生成树，全部 $n$ 个点的最小生成树有 $n-1$ 条边，那选 $k$ 个节点就是 $n-k$ 条边。

其次就是这个 No Answer 的问题，一共就 $m$ 条边，如果都选完了还没有选出 $n-k$ 条边来，那就是输出 No Answer 了。

代码

#include <bits/stdc++.h>
#define lowbit(x) x & (-x)
#define endl "\n"
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
namespace fastIO {
	inline int read() {
		register int x = 0, f = 1;
		register char c = getchar();
		while (c < '0' || c > '9') {
			if(c == '-') f = -1;
			c = getchar();
		}
		while (c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
		return x * f;
	}
	inline void write(int x) {
		if(x < 0) putchar('-'), x = -x;
		if(x > 9) write(x / 10);
		putchar(x % 10 + '0');
		return;
	}
}
using namespace fastIO;
int n, m, k, sum, ans, fa[10005];
struct node {
	int x, y, l;
}f[200005];
int find(int x) {return x == fa[x] ? x : fa[x] = find(fa[x]);}
bool cmp (node a, node b) {return a.l < b.l;}
int main() {
	cin >> n >> m >> k;
	for (int i = 1; i <= n; i ++) {
		fa[i] = i;
	}
	for (int i = 1; i <= m; i ++) {
		cin >> f[i].x >> f[i].y >> f[i].l;
	}
	sort (f + 1, f + m + 1, cmp);	
	for (int i = 1; i <= m; i ++) {
		if (find(f[i].x) != find(f[i].y)) {
			sum ++;
			fa[find(f[i].y)] = find(f[i].x);
			ans += f[i].l;	
		} 
		else continue;
		if (sum == n - k) {
			cout << ans << endl;
			return 0;
		}
	}
	cout << "No Answer" << endl;
	return 0;

}