Description
Farmer John已经决定把水灌到他的n(1<=n<=300)块农田,农田被数字1到n标记。把一块土地进行灌水有两种方法,从其他农田饮水,或者这块土地建造水库。 建造一个水库需要花费wi(1<=wi<=100000),连接两块土地需要花费Pij(1<=pij<=100000,pij=pji,pii=0). 计算Farmer John所需的最少代价。
Input
*第一行:一个数n
*第二行到第n+1行:第i+1行含有一个数wi
*第n+2行到第2n+1行:第n+1+i行有n个被空格分开的数,第j个数代表pij。
Output
*第一行:一个单独的数代表最小代价.
题解:
最小生成树,模型挺巧妙的.
设立超级源点 $S$.
对于每个点 $u$, 建立 $<S,u>$ 的边,边权为 $val_{u}$ ,代表该点要自己单独建立水库.
其次,再向图中与该点相连的边练边,边权按照输入来存.
跑一遍最小生成树即可.
这样就满足了图是联通的,每个点贡献可能是由超级源点给的,也可能是由其他点给的.
其次,再向图中与该点相连的边练边,边权按照输入来存.
跑一遍最小生成树即可.
这样就满足了图是联通的,每个点贡献可能是由超级源点给的,也可能是由其他点给的.
Code:
#include <bits/stdc++.h>
#define maxn 2000000
#define setIO(s) freopen(s".in","r",stdin)
using namespace std;
int A[maxn],u[maxn],v[maxn],val[maxn],cnt,w[maxn];
int cmp(int a,int b){ return val[a]<val[b]; }
struct U{
int p[maxn];
void init(){ for(int i=0;i<maxn;++i) p[i]=i; }
int find(int x){ return p[x]==x?x:p[x]=find(p[x]); }
void merge(int a,int b){
int x=find(a),y=find(b);
if(x==y) return;
p[x]=y;
}
}tree;
int main(){
// setIO("input");
tree.init();
int n;
scanf("%d",&n);
for(int i=1;i<=n;++i) ++cnt,scanf("%d",&val[cnt]),u[cnt]=0,v[cnt]=i;
for(int i=1;i<=n;++i)
for(int j=1,a;j<=n;++j) {
scanf("%d",&a);
if(i!=j) ++cnt,u[cnt]=i,v[cnt]=j,val[cnt]=a;
}
for(int i=1;i<=cnt;++i) A[i]=i;
sort(A+1,A+1+cnt,cmp);
int ans=0;
for(int i=1;i<=cnt;++i) {
int cur=A[i];
int a=u[cur],b=v[cur];
if(tree.find(a)!=tree.find(b)){
ans+=val[cur];
tree.merge(a,b);
}
}
printf("%d",ans);
return 0;
}