1601: [Usaco2008 Oct]灌水

题目链接

题目大意：n块农田灌水，可以修水库或者从其他农田引水，求最少代价。

题解：建个源点连边搞搞就行。很久以前的代码，码风诡异……

我的收获：……

#include <cstdio>#include <iostream>#include <algorithm>#define M 305using namespace std;int n,m,tot,x,y,z,ans,w;int v[M*M],u[M*M],val[M*M];int f[M],r[M*M];void init(){    cin>>n;    for(int i=1;i<=n;i++){        scanf("%d",&w);        u[++m]=n+1;v[m]=i;val[m]=w;    }    for(int i=1;i<=n;i++)    for(int j=1;j<=n;j++)    {        scanf("%d",&z);        if(i!=j){            u[++m]=i;v[m]=j;val[m]=z;        }    }}int cmp(const int a,const int b){    return val[a]<val[b];}int find(int x){    return f[x]==x?x:f[x]=find(f[x]);}void work(){    for(int i=1;i<=n;i++) f[i]=i;    for(int i=1;i<=m;i++) r[i]=i;    sort(r+1,r+1+m,cmp);    for(int i=1;tot<=n-1;i++)    {        int e=r[i];        x=find(u[e]),y=find(v[e]);        if(x!=y){            f[x]=y;tot++;            ans+=val[e];        }    }    printf("%d\n",ans);}int main(){     init();     work();     return 0;}