算法-程序设计课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)
东东为所有的田灌氵的最小消耗

Input

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

Output

东东最小消耗的MP值

Example
Input

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

Output

9

思路

如果只有一个源头，那我们计算最小生成树就行了，但是这里出现了多个源头，如何对源头进行选择？对每个源头都算一个生成树吗？这会超时，我们要想出更巧妙的解决方案才行。

解决方案很简答，将“天”视作一个节点，所有能浇灌的农田都与“天”相连，我们最终只需求求以“天”为源头的最小生成树就好了。

1. 根据输入求

心得

最近做了一道百度笔试题，和这题很相似。

最小生成树的问题一般不仅仅是考算法，还要考问题的转化，比如这道题考如何考虑多个“源头”的问题，将源头虚拟为一个总节点即可。
百度那道笔试题也是如此，将宇宙空间边界视作一个节点。

#include <stdio.h>

#include <algorithm>
#include <iostream>
using namespace std;

//代表路径消耗的结构体
struct path {
	int from, to, cost;
	bool operator<(const path &no) const {
		return cost < no.cost;
	}
};

path k[100010];

int n, vv, uu, father[1000], w, p, id = 0, sum = 0;

int find(int x) {
	return father[x] = x == father[x] ? x : find(father[x]);
}

int kruskal() {
	//初始化并查集
	for (int i = 1; i <= n + 1; i++) father[i] = i;

	id = 0;
	for (int i = 1; i <= n * n; i++) {
		uu = find(k[i].from);
		vv = find(k[i].to);
		if (uu != vv) {
			father[vv] = uu;
			sum += k[i].cost;
			if (++id == n)
				return sum;
		}
	}
}
int main() {
	scanf("%d", &n);  //获得田数
	id = 0;

	//从“天”到田地的消耗
	for (int i = 1; i <= n; i++) {
		++id;
		scanf("%d", &k[id].cost);
		k[id].from = n + 1;
		k[id].to = i;
	}

	//获取所有田地之间建立通道的消耗
	for (int from = 1; from <= n; from++) {
		for (int to = 1; to <= n; to++) {
			int cost;
			scanf("%d", &cost);
			if (from != to) k[++id] = {from, to, cost};
		}
	}

	sort(k, k + n * n);
	cout << kruskal() << endl;
	return 0;
}