掌握魔法的东东——灌溉田（最小生成树）

问题描述

东东在老家农村无聊，想种田。农田有 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 个点连边为 Wi，然后对这n+1 个点跑最小生成树即可。
参考代码如下：

#include<stdio.h>
#include<algorithm>
using namespace std;
struct edge{
	int e1,e2,w;
	bool operator<(const edge&x){return w<x.w;}
}a[91000];
int fa[310];
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
int Kruskal(int n){
	int E=0,i=0,ans=0;
	for(int i=0;i<=n;i++)fa[i]=i;
	while(E<n){
		int u = find(a[i].e1), v = find(a[i].e2);
		if (u != v){
			fa[u] = v;
			E++;
			ans+=a[i].w;
		}
		i++;
	}
	return ans;	
}
int main(){
	int n,e,x;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&x);
		a[i].e1=i+1;
		a[i].e2=0;
		a[i].w=x;
	}
	e=n;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++){
			scanf("%d",&x);
			if(i<j){
				a[e].e1=i;
				a[e].e2=j;
				a[e].w=x;
				e++;
			}
		}
	sort(a,a+e);
	printf("%d",Kruskal(n));
	return 0;
}