题意
东东在老家农村无聊,想种田。农田有 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
思路
对于为每块地降雨的MP消耗看作是一块0号田地到该田地引流的MP消耗,因此将田地的数量看作是n+1块。然后将此问题建模成为一个加权无向图的生成树问题,使用kruskal算法,在所有的加权边中,找到权重最小的边,如果加入选中该边之后形成环(使用并查集,如果该边的两个顶点已经在两个同一个并查集中,则为形成环),则将此边删除,之后重复上述过程,直到将所有的边遍历完成。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=4e2+10;
int a[maxn],result=0;
void ini(){for(int i=0;i<maxn;i++) a[i]=i;}
struct edge
{
int u,v,mp;
//edge *next;
bool operator<(edge e){return mp<e.mp;}
bool operator>(edge e){ return mp>e.mp;}
}e[(maxn+1)*maxn/2];
int find(int x)
{
int y=x;
while(x!=a[x])
x=a[x];
while(y!=x)
{
int temp=y;
y=a[y];
a[temp]=x;
}
return x;
}
void unite(int x,int y)
{
int temp=find(y);
a[temp]=find(x);
}
int main()
{
ini();
int n,count=0;
cin>>n;
for(int i=1;i<=n;i++)
{
e[i-1].u=0;
e[i-1].v=i;
int temp=0;
cin>>temp;
e[i-1].mp=temp;
}
count=n;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
int temp=0;
cin>>temp;
if(i<j)
{
e[count].u=i;
e[count].v=j;
e[count].mp=temp;
count++;
}
}
sort(e,e+count);
for(int i=0;i<count;i++)
{
int u=e[i].u,v=e[i].v,mp=e[i].mp;
if(find(u)!=find(v))
{
unite(u,v);
result+=mp;
}
}
cout<<result<<endl;
return 0;
}