程序设计Week6——C-掌握魔法的东东Ⅰ

题目描述

东东在老家农村无聊，想种田。农田有 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值

解题思路

这道题乍一看就是用最小生成树求解，但是有一个关键问题，就是那句“黄河之水天上来”，因为灌溉肯定需要水，但所有田地一开始都没有水，必须从天上引水，可是如何引水，给哪些田地引水，引几块田地，似乎毫无思路。
巧妙的一个点是：把”天”考虑为一个有水源的田地，可以考虑其编号为0，那所有的的所谓天上来的消耗，就可以理解为连通0和i块田的消耗，这样，我们只需要找到n次连接（因为0号田也需要被连接），使得整个田地构成一个连通图，即n+1个点的最小生成树。
由此先建立边的结构体，根据输入构建边数组，并根据每条边的消耗，即边权，进行排序，利用Kruskal算法求解最小生成树，并计算总消耗，最后输出即可。
Kruskal算法：

• 以边为单位，每次选择边权最小的边，加入最小生成树集合；
• 选择时，需要避免出现环，做法是利用并查集，每加入一条边，就将边上对应的点加入这棵生成树对应的集合内，再加入下一条边时，判断这条边的两个端点是否再一个集合内，若不在，则说明这条边可选，加入生成树边集内，并将两个点合并到一个集合（生成树点集）；若在，则说明该边不可选，跳过判断下一条边；
• 直到选出n条边（因为有n个点），跳出循环；若遍历结束后依然没有选出n条边，则说明无法生成。

实现代码

#include<iostream>
#include<algorithm>
using namespace std;

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

edge e[100000];
int size=0;				//边的个数 
int f[500];		//f存储每个点所在树的代表元 

int find(int u)
{//找到点u所在树的代表元 
	if(f[u]==u)
		return u;
	return f[u]=find(f[u]);
}

bool combine(int u,int v)
{//合并树，将小树合并到大树上 
	int x=find(u),y=find(v);
	if(x==y)
		return false;
	f[x]=y;
	return true;
}

int main()
{
	int n;
	cin>>n;
	for(int i=0;i<500;++i)
	{//初始化 
		f[i]=i;
	}
	for(int i=1;i<=n;++i)
	{
		e[size].u=0;
		e[size].v=i;
		cin>>e[size].w;
		size++;
	}
	for(int i=1;i<=n;++i)
	{
		for(int j=1;j<=n;++j)
		{
			int p;
			cin>>p;
			if(p!=0)
			{
				e[size].u=i;
				e[size].v=j;
				e[size].w=p;
				size++;
			}
		}
	}
	sort(e,e+size);
	int i=0,sum=0;		//i是当前选择的边数，sum是当前消耗的mp 
	for(int j=0;j<size;++j)
	{
		if(combine(e[j].u,e[j].v))
		{
			sum+=e[j].w;
			i++;
			if(i==n)
				break;
		}
	}
	cout<<sum<<endl;
	return 0;
}

总结

本题核心是Kruskal最小生成树算法，但最重要的一点是，如何处理“黄河之水天上来”这个问题，巧妙的将天设为一个超级源点，也当作图的一个节点来考虑，整个问题就变得很简单了，只需要套板子跑就行。因此这种图的问题，如何将已知的题目信息转化为显式的图的关系是一个重点，也是需要不断提升的一个能力。