BZOJ_1601_[Usaco2008_Oct]_灌水_(最小生成树_Kruskal)

描述

---

http://www.lydsy.com/JudgeOnline/problem.php?id=1601

有\(n\)个田地需要灌溉,每个田地可以自己引水,花费为\(w[i]\),或者连接其他被灌溉的田地,花费为\(p[i][j]\),求最小花费.

 

分析

---

我第一眼看以为是dp,发现不对...

如果田地不能自己引水只能和其他田地连接,就是裸的最小生成树.

这里加了一个可以自己引水,相当于加了一个和所有点都有连边,且权值分别为\(w[i]\)的结点,然后求\(n+1\)个点的最小生成树即可.

 

 

#include <bits/stdc++.h>
using namespace std;

const int maxn=300+5;
struct edge{
    int u,v,w;
    edge(int u=0,int v=0,int w=0):u(u),v(v),w(w){}
    bool operator < (const edge &a) const { return w<a.w; }
}g[maxn*maxn];
int n,ect,ans;
int f[maxn];
inline int read(int &x){ x=0;int k=1;char c;for(c=getchar();c<'0'||c>'9';c=getchar())if(c=='-')k=-1;for(;c>='0'&&c<='9';c=getchar())x=x*10+c-'0';return x*=k; }
inline int find(int x){ return x==f[x]?x:f[x]=find(f[x]); }
int main(){
    read(n);
    for(int i=1,w;i<=n;i++) g[++ect]=edge(0,i,read(w)), f[i]=i;
    for(int i=1;i<=n;i++)for(int j=1,w;j<=n;j++){
        read(w);
        if(j<i) g[++ect]=edge(i,j,w);
    }
    sort(g+1,g+ect+1);
    for(int i=1,j=n;i<=ect&&j;i++){
        int fu=find(g[i].u),fv=find(g[i].v);
        if(fu!=fv) f[fu]=fv, j--, ans+=g[i].w;
    }
    printf("%d\n",ans);
    return 0;
}