LOJ10066 新的开始

LOJ10066 新的开始

prim

典型题。碰到这种情况，只要建一个虚拟节点，和其他的点连边，按题目给权值即可

代码中把n+1当成虚拟节点

懒得写kruskal就用prim了

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,ans,a[][],d[];
bool vis[];
int main(){
    memset(d,,sizeof(d));
    scanf("%d",&n);
    for(int i=;i<=n;++i) scanf("%d",&a[i][n+]),a[n+][i]=a[i][n+];
    for(int i=;i<=n;++i)
        for(int j=;j<=n;++j){
            scanf("%d",&a[i][j]);
            if(a[i][j]==) a[i][j]=d[]; //改成极大值
        }
    a[n+][n+]=; d[]=;
    for(int i=;i<=n+;++i){ //普通的prim
        int u=;
        for(int j=;j<=n+;++j) if(d[j]<d[u]&&!vis[j]) u=j;
        vis[u]=; ans+=d[u];
        for(int j=;j<=n+;++j) if(a[u][j]<d[j]&&!vis[j]) d[j]=a[u][j];
    }
    printf("%d",ans);
    return ;
}