洛谷——买礼物

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=510,maxm=300010;
struct edge{int f,t,c;}e[maxm];
bool cmp(edge a,edge b){return a.c<b.c;}
int fa[maxn],n,m=0,a,ans=0;
void init(){for(int i=0;i<=n;i++)fa[i]=i;}
int find(int x){
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
void unite(int x,int y){
    x=find(x);y=find(y);
    if(x==y)return;
    fa[x]=y;
}
void kru(){
    init();sort(e+1,e+m+1,cmp);
    for(int i=1;i<=m;i++){
        if(find(e[i].f)==find(e[i].t))continue;
        unite(e[i].f,e[i].t);ans+=e[i].c;
    }
}
int main()
{
    scanf("%d%d",&a,&n);int x;
    for(int i=1;i<=n;i++){
        e[++m].f=0;e[m].t=i;e[m].c=a;
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            scanf("%d",&x);
            if(!x||i<=j)continue;
            e[++m].f=i;e[m].t=j;e[m].c=x;
        }
    }
    kru();printf("%d\n",ans);
    return 0;
}