洛谷 - P1194 买礼物

早过了，但O2会爆
于是进行面向题解的编码
裸kruskal

#include<bits/stdc++.h> 
using namespace std;
struct node {
	int x,y,s;
} d[200000];
int n,m,ans,num,f[1000],w,k,c;
bool ok[1000];
bool cmp(node a,node b){ 
	return a.s<b.s;
}
int gf(int x){ 
	if(f[x]==x)return x;
	return f[x]=gf(f[x]);
}
void kruskal(){ 
	int x,y;
	for(int i=1;i<=n;i++)f[i]=i;
	for(int i=1;i<=k-num;i++){
		x=gf(d[i].x);
		y=gf(d[i].y);
		if(x!=y){
			f[x]=y;
			c++;
			ans+=d[i].s;
		}
	}
}
int main(){
	cin>>m>>n;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++){
			cin>>w;
			if(i>=j||w==0)continue;
			d[++k].x=i;
			d[k].y=j;
			d[k].s=w;
			if(d[k].s>m)num++;
		}
	sort(d+1,d+k+1,cmp); 
	kruskal();
	if(c==n-1)cout<<ans+m;
	else cout<<ans+(n-c)*m;
	return 0;
}