LOJ10066

LOJ10066

将建井看成是向“0”号节点连边，然后正常写最小生成树就好了。。
稠密图，PRIM快

#include<bits/stdc++.h>
#define gt() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++)
using namespace std;
static char buf[1000000],*p1=buf,*p2=buf;
const int maxn=305,INF=1e9;
int n,Ans,cst[maxn][maxn],dst[maxn];bool vis[maxn];
int read(){
    int ret=0;char ch=gt();
    while(ch<'0'||ch>'9') ch=gt();
    while(ch>='0'&&ch<='9') ret=ret*10+ch-'0',ch=gt();
    return ret;
}
int main(){
    n=read();
    for(int i=1;i<=n;i++) dst[i]=read();
    for(int i=1;i<=n;i++)
      for(int j=1;j<=n;j++) cst[i][j]=read();
    for(int i=1;i<=n;i++){
        int min_x=INF,k;
        for(int j=1;j<=n;j++)if(!vis[j]&&dst[j]<min_x) min_x=dst[j],k=j;
        Ans+=min_x,vis[k]=1;
        for(int j=1;j<=n;j++)if(!vis[j]&&cst[k][j]<dst[j]) dst[j]=cst[k][j];
    }
    printf("%d\n",Ans);
    return 0;
}