Week6--作业 -- C -- 掌握魔法の东东 I[生成树]

题目描述

东东在老家农村无聊，想种田。农田有 n 块，编号从 1~n。种田要灌氵
众所周知东东是一个魔法师，他可以消耗一定的 MP 在一块田上施展魔法，使得黄河之水天上来。他也可以消耗一定的 MP 在两块田的渠上建立传送门，使得这块田引用那块有水的田的水。 (1<=n<=3e2)
黄河之水天上来的消耗是 Wi，i 是农田编号 (1<=Wi<=1e5)
建立传送门的消耗是 Pij，i、j 是农田编号 (1<= Pij <=1e5, Pij = Pji, Pii =0)
东东为所有的田灌氵的最小消耗

输入

第1行：一个数n
第2行到第n+1行：数wi
第n+2行到第2n+1行：矩阵即pij矩阵

输出

第1行：一个数n
第2行到第n+1行：数wi
第n+2行到第2n+1行：矩阵即pij矩阵

样例输入

4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0

样例输出

9

思路

综述

这道题主要考察了一个新的算法：Kruskal
参考：图论、并查集。

Kruskal

Step1：
边集排序

	std::sort(e, e + tot, cmp);

Step2：
从最小的开始遍历边集；
如果顶点个数集为n，退出；

		if (unit(e[i].u, e[i].v)) {
			total += e[i].w;
			num++;
			if (num == n)break;
		}

本题

所有的农田看成点，连接之间有边，天空看做一个点，该点与所有的农田点都有边相连。所以该题就成了n+1个点的最小生成树问题；

过程

Step1:输入

注意一点，如果输入为0则不加入到边集内
输入超级源点

		if (w != 0) {
			term.u = n;
			term.v = i;
			term.w = w;
			e[tot] = term;
			tot++;
		}

输入农田点
注意一点：因为这是无向图，所以只需要记录一半即可；
可以这样处理：只记录(i,j)中，i>j的情况；

	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			cin >> w;
			if (i < j)continue;
			if (w == 0)continue;
			term.u = i;
			term.v = j;
			term.w = w;
			e[tot] = term;
			tot++;
		}
	}

Step2:跑Kruskal

具体过程见上面

int kruskal() {
	std::sort(e, e + tot, cmp);
	int total = 0;
	int num = 0;
	for (int i = 0; i < tot; i++) {
		if (unit(e[i].u, e[i].v)) {
			total += e[i].w;
			num++;
			if (num == n)break;
		}
	}
	return total;
}	

总结

Kruskal算法用到两个知识：并查集和Kruskal算法本身；
图论的问题，常常需要我们从长长的文字中抽象出图，然后再根据所学解决问题；

代码

上文有详细注释

#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e5 + 5;
int n;
int tot = 0;
struct Edge {
	int u, v, w;
	bool operator < (Edge& t) {
		return w < t.w;
	}
}e[maxn];
int par[maxn];
Edge term;
void init() {
	for (int i = 0; i < maxn; i++) {
		par[i] = i;
	}
}
int find(int x) {
	if (par[x] == x)return x;
	else return par[x] = find(par[x]);
}
bool unit(int ax, int ay) {
	int x = find(ax);
	int y = find(ay);
	if (x == y)return false;
	par[x] = par[y];
	return true;
}
bool cmp(Edge a, Edge b) { return a.w < b.w; }
int kruskal() {
	std::sort(e, e + tot, cmp);
	int total = 0;
	int num = 0;
	for (int i = 0; i < tot; i++) {
		if (unit(e[i].u, e[i].v)) {
			total += e[i].w;
			num++;
			if (num == n)break;
		}
	}
	return total;
}

int main() {
	cin >> n;
	int w;
	init();
	for (int i = 0; i < n; i++) {
		cin >> w;
		if (w != 0) {
			term.u = n;
			term.v = i;
			term.w = w;
			e[tot] = term;
			tot++;
		}
	}
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			cin >> w;
			if (i < j)continue;
			if (w == 0)continue;
			term.u = i;
			term.v = j;
			term.w = w;
			e[tot] = term;
			tot++;
		}
	}
	cout << kruskal() << endl;
}