bzoj 1601: [Usaco2008 Oct]灌水 （最小生成树）

题目描述

传送门

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

题解

等价于把所有的点分成好多连通块，使连通块中的点连通且选出一个建造水库。
连通的代价最小感觉好像最小生成树啊。
新加节点，所有点向新加的点连边，边权使建水库的代价，做最小生成树即可。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#define N 200000
using namespace std;
int cnt,n,fa[N];
struct data{
    int x,y,val;
}e[N];
int cmp(data a,data b)
{
    return a.val<b.val;
}
int find(int x)
{
    if (fa[x]==x) return x;
    fa[x]=find(fa[x]);
    return fa[x];
}
int main()
{
    freopen("a.in","r",stdin);
    scanf("%d",&n);
    for (int i=1;i<=n;i++) {
        int x; scanf("%d",&x);
        e[++cnt].x=0; e[cnt].y=i; e[cnt].val=x;
    }
    for (int i=1;i<=n;i++) 
     for (int j=1;j<=n;j++) {
        int x; scanf("%d",&x);
        if (j>=i) continue;
        e[++cnt].x=i; e[cnt].y=j; e[cnt].val=x;
     }
    sort(e+1,e+cnt+1,cmp);
    int ans=0,size=0;
    for (int i=0;i<=n;i++) fa[i]=i;
    for (int i=1;i<=cnt;i++) {
        int r1=find(e[i].x); int r2=find(e[i].y);
        if (r1!=r2) {
            fa[r2]=r1;
            size++;
            ans+=e[i].val;
        }
        if (size==n) break;
    }
    printf("%d\n",ans);
}