bzoj1601 [Usaco2008 Oct]灌水(prim)

直接做不太好做，我们搞一个超级源点，让它向每一个点建一条权值为w的边，然后我们求最小生成树即可。这样可以保证至少有一个点我们建了水库。而且所有点连通。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define pa pair<int,int>
#define N 310
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,a[N][N],d[N],ans=0;
bool vis[N];
void Prim(){
    priority_queue<pa,vector<pa>,greater<pa> >q;memset(d,0x3f,sizeof(d));
    d[0]=0;q.push(make_pair(0,0));
    while(!q.empty()){
        int x=q.top().second;q.pop();if(vis[x]) continue;
        vis[x]=1;ans+=d[x];
        for(int i=1;i<=n;++i)
            if(a[x][i]&&a[x][i]<d[i]) 
                d[i]=a[x][i],q.push(make_pair(d[i],i));
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();
    for(int i=1;i<=n;++i) a[i][0]=a[0][i]=read();
    for(int i=1;i<=n;++i)
        for(int j=1;j<=n;++j) a[i][j]=read();
    Prim();
    printf("%d\n",ans);
    return 0;
}