WA on #2 90分求助 悬赏一个关注
查看原帖
WA on #2 90分求助 悬赏一个关注
481527
AC_CSP楼主2023/2/10 18:15
#include<bits/stdc++.h>
using namespace std;
const int N=3e2+7;
const int M=9e4+7;
struct node{
	int u,v,w;
}G[M];
int cnt;
int n,m,tmp,ans;
int fa[N];
inline bool cmp(node x,node y){
	return x.w<y.w;
}
inline int find(int x){
	if(x!=fa[x]) return fa[x]=find(fa[x]);
	return x;
}
int main(){
	scanf("%d",&n);m=n+n*n;
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=n;i++) scanf("%d",&tmp),G[++cnt]=(node){i,n+1,tmp};
	for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) scanf("%d",&tmp),G[++cnt]=(node){i,j,tmp};
	sort(G+1,G+m+1,cmp);
	for(int i=1;i<=m;i++){
		int x=find(G[i].u),y=find(G[i].v);
		if(x!=y) fa[y]=x,ans+=G[i].w;
	}
	printf("%d\n",ans);
	return 0;
}
2023/2/10 18:15
加载中...