BZOJ 1601 USACO 2008 Oct. 灌水

【Description】

Farmer John已经决定把水灌到他的n(1<=n<=300)块农田，农田被数字1到n标记。把一块土地进行灌水有两种方法，从其他农田饮水，或者这块土地建造水库。 建造一个水库需要花费wi(1<=wi<=100000),连接两块土地需要花费Pij(1<=pij<=100000,pij=pji,pii=0). 计算Farmer John所需的最少代价。

【题解】

新建一个节点0，把各个点与0点连边，边权为在该点上建水库的代价

然后跑一边最小生成树就行了

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=400,maxm=200010;
int n,fa[maxn],tot=0,num=0;
long long ans=0;
struct edge{int x,y,w;}e[maxm];
void read(int &k){
    k=0; int f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    k*=f;
}
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
bool cmp(edge a,edge b){return a.w<b.w;}
int main(){
    read(n); for (int i=0;i<=n;i++) fa[i]=i;
    for (int i=1;i<=n;i++)e[++tot].x=i,e[tot].y=0,read(e[tot].w);
    for (int i=1;i<=n;i++)
    for (int j=1;j<=n;j++)e[++tot].x=i,e[tot].y=j,read(e[tot].w);
    sort(e+1,e+1+tot,cmp);
    for (int i,x,y=1;i<=tot;i++){
        if (find(x=e[i].x)!=find(y=e[i].y)){
            ans+=e[i].w; fa[find(x)]=find(y); num++;
        }
        if (num==n) break;
    }
    return printf("%lld\n",ans),0;
}