【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)

东东为所有的田灌水的最小消耗

输入说明

第1行：一个数n

第2行到第n+1行：数wi

第n+2行到第2n+1行：矩阵即pij矩阵

输出说明

东东最小消耗的MP值

Sample Input

4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0

Sample Output

9

思路

将每块农田想象成一个点，而两点之间引水造成的MP消耗可看作两点之间的加权边，这样农田便成为一个加权图。接下来的问题是如何处理天上引水这一消耗，我们将天看作一个超级原点，它也有到农田各点的加权边，即引水的MP消耗。要使得可以灌溉所有农田而且消耗最少，该问题可转化为一个在加权图中寻找最小生成树的问题，因此接下来的问题便是如何在我们创建的图中找到它的最小生成树。
这里采用Kruskal算法，根据权值对所有的边进行排序，按权值升序来考察各边，若边的端点不在同一个集合中，便合并这两点所在集合（可采用并查集），同时选中该边，直到选中n条边（有n+1个点），这些选中的边便可组成该图的最小生成树，其权值相加便为最小的MP消耗。

代码

#include <iostream>
#include<algorithm>
using namespace std;
const int size=1e5;
struct Edge{
	int s,t,value;
	bool operator<(Edge&b){
		return value<b.value;
	}
}edge[size];
int par[size];
void ini(int n)
{
	for(int i=0;i<=n;i++)
		par[i]=i;	
}
int find(int n)
{
	return par[n]==n?n:par[n]=find(par[n]);
}
bool unite(int a,int b)
{
	a=find(a),b=find(b);
	if(a==b)return false;
	par[b]=a;
	return true;
}
int main(int argc, char** argv) {
	int n;
	scanf("%d",&n);
	ini(n);
	int number=0; 
	for(int i=1;i<=n;i++)
	{
		int temp;
		scanf("%d",&temp);
		edge[number]={0,i,temp};
		number++;
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			int temp;scanf("%d",&temp);
			if(i==j)continue;
			edge[number]={i,j,temp};
			number++;
		}
	}
	sort(edge,edge+number);
	int wp=0,pickedge=0;
	for(int i=0;i<number;i++)
	{
		if(unite(edge[i].s,edge[i].t))
		{
			wp+=edge[i].value;
			pickedge++;
			if(pickedge==n)
				break;
		}
	}
	printf("%d\n",wp);
	return 0;
}