bzoj1601 [Usaco2008 Oct]灌水

题目

构造一个虚拟点0，连向每一个点，边权为建造水站的费用，之后各个之间连边，跑最小生成树就OK了。

#include<bits/stdc++.h>
#define N 500
using namespace std;
int n,f[N+1],x,tot,ans,fx,fy;
struct edge{
    int x;
    int y;
    int val;
    bool operator <(const edge &A) const
    {
        return val<A.val;
    }
    void insert(int a,int b,int c)
    {
        x=a,y=b,val=c;
    }
};edge A[N*N+1];
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    int x=0,b=1;
    char c=nc();
    for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
    for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
    return x*b;
}
inline void write(int x)
{
    if(x==0)putchar('0');
    else
    {
        char buf[15];
        int len=0;
        if(x<0)putchar('-'),x=-x;
        while(x)buf[++len]=x%10+'0',x/=10;
        for(int i=len;i>=1;i--)putchar(buf[i]);
    }
    putchar(' ');
}
inline int find(int x)
{
    return x==f[x]?f[x]:f[x]=find(f[x]);
}
int main()
{
    freopen("in.txt","r",stdin);
    n=read();
    for(int i=0;i<=n;i++)f[i]=i;
    for(int i=1;i<=n;i++)
        x=read(),A[++tot].insert(0,i,x);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            x=read(),A[++tot].insert(i,j,x);
    sort(A+1,A+tot+1);
    for(int i=1;i<=tot;i++)
    {
        int fx=find(A[i].x),fy=find(A[i].y);
        if(fx==fy)continue;
        f[fx]=fy;
        ans+=A[i].val;
    }
    cout<<ans;
    return 0;
}