58WA求调
查看原帖
58WA求调
794148
cyx20091026楼主2024/9/19 19:21
#include <bits/stdc++.h>
using namespace std;
int n,m;//原价n,m样东西 
int s;//边数 
int ans,cnt;
struct edge{
	int front,nxt,len;
}a[2000010];
int f[100001];
bool cmp(edge x,edge y){
	return x.len<y.len;
}
int find(int x){
	return f[x]==x?x:find(f[x]);
}
void kruskal(){
	sort(a+1,a+1+s,cmp);
	for(int i=1;i<=s;i++){
		int eu=find(a[i].front);
		int ev=find(a[i].nxt);
		if(eu==ev) continue;
		ans+=a[i].len;
		cnt++;
		f[eu]=ev;
		if(cnt==m-1) return;
	}
}
int main(){
	cin>>n>>m;
	s=m*m;
	for(int i=1;i<=m;i++){
		for(int j=1;j<=m;j++){
			int w=i*j;
			cin>>a[w].len;
			if(a[w].len==0||a[w].len>n){
				a[w].len=n;
			}
			a[w].front=i,a[w].nxt=j;
		}
	}
	for(int i=1;i<=m;i++){
		f[i]=i;
	}
	kruskal();
	cout<<ans+n;
	return 0;
}
2024/9/19 19:21
加载中...