最小生成树,解题的关键是要把所有可能建水库的点与0点(自己设的一个原点)之间建边,建水库的费用作为这条边的权值,这样这个设的0点就是超级源,然后就MST就可以啦,哈哈哈!
#include<iostream>
#include<algorithm>
using namespace std;
struct node{
int x,y,d;
bool operator<(const node &b)const{
return d<b.d;
}
}e[100010];
int father[510];
int find(int x){
while(father[x]!=x){
x=father[x];
}
return x;
}
bool combine(int l,int r){
int l1=find(l);
int r1=find(r);
if(l1!=r1){
father[r1]=l1;
return true;
}
return false;
}
int main(){
int n;
cin>>n;
int flag=0;
for(int i=1;i<=n;i++){
e[flag].x=0;e[flag].y=i;
cin>>e[flag].d;flag++;
father[i]=i;
}
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
e[flag].x=i;e[flag].y=j;
cin>>e[flag].d;flag++;
}
}
sort(e,e+flag);
int ans=0;
for(int i=0;i<flag;i++){
if(combine(e[i].x,e[i].y)){
ans+=e[i].d;
}
}
cout<<ans<<endl;
}