[Usaco2008 Oct]灌水(MST)

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

思路：加一个超级源点，然后跑一遍MST。作为一个经典的题纪录一下。

#include <bits/stdc++.h>
using namespace std;
 
typedef long long ll;
const int INF=0x3f3f3f3f;
const int maxn=305;
int T,n,m;
int w[maxn];
int G[maxn][maxn];
int low[maxn],vis[maxn];
 
void Prim(){
    memset(vis,0,sizeof(vis));
    int pos=0,min,res=0;
    for(int i=0;i<=n;i++) low[i]=G[pos][i];
    vis[pos]=1;
    for(int i=0;i<n;i++){
        min=INF;
        for(int j=0;j<=n;j++){
            if(!vis[j]&&min>low[j]){
                min=low[pos=j];
            }
        }
        res+=min;
        vis[pos]=1;
        for(int j=0;j<=n;j++){
            if(!vis[j]&&low[j]>G[pos][j]){
                low[j]=G[pos][j];
            }
        }
    }
    printf("%d\n",res);
}
 
 
int main() {
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++){
            scanf("%d",&w[i]);
            G[0][i]=w[i];
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                scanf("%d",&G[i][j]);
            }
        }
        Prim();
    }
    return 0;
}