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 个点连边为Wi，然后对这 n+1 个点跑最小生成树即可

kruskal 求最小生成树

解法

step0 建立edge结构体存储边的两个顶点和权重，重载小于号为比较权重的大小
step1 将边录入到Edges结构体数组中，按照边权重进行从小到大的排序，不断取出权重最小的边，用并查集判断加入此边之后是否会形成环，在不形成环时，加入该边，当边数为点数-1时停止取出边。此时得到kruskal生成的最小生成树的权重和，否则不能得到最小生成树。

代码

#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;
int n;
//并查集
int par[maxn];
void init(int n)
{
	for(int i=0;i<=n;i++)
	par[i]=i;
} 
int find(int x)
{
	return par[x]==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,Es+tot);
	int cnt=0,ans=0;
	for(int i=0;i<tot;i++)
		if(unite(Es[i].u,Es[i].v))
		{
			ans+=Es[i].w;
			if(++cnt==n) return ans;
//			cout<<Es[i].u<<' '<<Es[i].v<<' '<<Es[i].w<<endl;
		}
	return -1;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&Es[tot].w);
		Es[tot].u=0;
		Es[tot].v=i;
		tot++;
	}
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			scanf("%d",&Es[tot].w);
			if(i>=j)//注意这里不是等于，因为一条边尽可存一次
				continue;
			Es[tot].u=i;
			Es[tot].v=j;
			tot++;
		}
		printf("%d",kruskal());
	return 0;
}