Week6-C、D 最小生成树

掌握魔法の东东 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值

Input
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0
Output
9

问题分析

如果把“黄河之水天上来”删掉，那么就是一个最小生成树问题。
考虑图重构：加一个超级源点0 号，并向n 个点连边为Wi，然后对这 n+1 个点跑最小生成树即可。

最小生成树常用算法有kruskal和prim，本题中选择用kruskal。只存储边集，按照权重大小排序，每次都选择权最小且不会连通的边加入到树中，加上这条边的权重作为生成树的权重。

判断会不会连通可以选择用并查集，祖先不同则不会连通。当选中n条边，则生成了最小生成树，返回权重。如果最后遍历完所有的边仍然未曾选出n条边，则无法生成最小生成树。

注意
边集数组一定一定一定要开的足够大！！！

代码实现

#include<stdio.h>
#include<algorithm>
using namespace std;

struct edge
{
   
	int u,v;
	int p;
	bool operator<(const edge &b) const
	{
   
		return p < b.p;
	}
};


int n;
edge edg[100000];
int f[350];
int index=0;

int find(int i)
{
   
	if(f[i] == i)
	    return i;
	else
	    return find(f[i]);
}

bool unite(int a, int b)
{
   
	int x=find(a);
	int y=find(b);
	if(x == y) return false;
	else{
   
		f[x] = y<