WEEK 6 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

思路：

将每块农田看作一个点，在两点之间引水即将两个点连接起来，两点之间引水的MP消耗可看作此边的权值，这样整片农田便成为了一个加权图。我们将天看作一个超级原点，它也有到农田各点的加权边。这样，该问题可转化为图的重构的问题，即在一个加权图中找到最小生成树。
这里采用Kruskal算法，根据权值将所有的边按升序排序，使用并查集遍历所有的边，若边有端点不在同一个集合中，便合并这两点所在集合，同时选中该边，直到选中n条边（有n+1个点），这些选中的边便可组成该加权图的最小生成树，其权值相加便为答案。

代码：

#include <iostream>
#include <algorithm>

using namespace std;

#define rep(i,s,t)for(int i=s;i<=t;i++)

const int maxn = 3e2 + 10;
int par[maxn];
int n, tot;

struct edge
{
	int u, v, w;
	bool operator<(const edge &e)const
	{
		return w < e.w;
	}
}es[maxn*maxn];

void init(int n)
{
	for (int i = 0; i <= n; i++)
		par[i] = i;
}
int find(int x)
{
	if (par[x] == x)
		return x;
	else
		return par[x] = find(par[x]);
}
bool unite(int x, int y)
{
	x = find(x), y = find(y);
	if (x == y)
		return false;
	par[x] = y;
	return true;
}

int kruskal()
{
	init(n);
	sort(es, es + tot);
	int cnt = 0, ans = 0;
	rep(i, 0, tot - 1)if (unite(es[i].u, es[i].v))
	{
		ans += es[i].w;
		if (++cnt == n)
			return ans;
	}
	return -1;
}

int main()
{
	tot = 0;
	scanf_s("%d", &n);
	rep(i, 1, n)scanf_s("%d", &es[tot].w), es[tot].u = 0, es[tot].v = i, tot++;
	rep(i, 1, n)rep(j, 1, n)
	{
		scanf_s("%d", &es[tot].w);
		if (i == j)
			continue;
		es[tot].u = i, es[tot].v = j, tot++;
	}
	printf("%d", kruskal());
	return 0;
}