题目描述
东东在老家农村无聊,想种田。农田有 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最小生成树算法,但最重要的一点是,如何处理“黄河之水天上来”这个问题,巧妙的将天设为一个超级源点,也当作图的一个节点来考虑,整个问题就变得很简单了,只需要套板子跑就行。因此这种图的问题,如何将已知的题目信息转化为显式的图的关系是一个重点,也是需要不断提升的一个能力。