程序设计思维与实践 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
解题思路：
建立一个零号节点，与各块田连边。至少有一条边是0号点与其中一块田。这样就变成了求n+1个结点图的最小生成树，用kruskal算法即可，每次选边值最小的边进入图，用并查集判断此边的两个端点是否在一个集合中。

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=3e2+5;

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


int par[maxn];
void init(int n) {
	for(int i=0;i<=n;i++)
	   par[i]=i;
}
int find(int x) {
	return x==par[x]?x:(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+1,es+tot+1);
	int num=0,ans=0;
	for(int i=1;i<=tot;i++)
	{
		if(unite(es[i].u,es[i].v))
		{
			ans+=es[i].w;
			num++;
		
	    }	
		if(num==n)   return ans;
	}
	return -1;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)   
	{
	   int x;
	   scanf("%d",&x);
	es[++tot].u=0;
	es[tot].v=i;
	es[tot].w=x;
     }
	for(int i=1;i<=n;i++)
	{
	  for(int j=1;j<=n;j++)
	  {
	  	int x;
	  	scanf("%d",&x);
	  	if(i>j)
	  	es[++tot].u=i,es[tot].v=j,es[tot].w=x; 
	   }
    }
	printf("%d\n",kruskal());
	return 0; 
}