我和标程差哪?
查看原帖
我和标程差哪?
128241
XueYunAW楼主2022/2/26 19:44

我的:

#include<cstdio>
#include<algorithm>
#define re register
int n,m,l,ans,f[50*99+1];
struct information{
	int x,y,num;
}tot[50*99+1];
bool cmp(information a,information b){
	return a.num<b.num;
}
bool find(int x){
	if(x==f[x])
	return x;
	f[x]=find(f[x]);
	return f[x];
}
int main(){
	scanf("%d",&n);
	for(re int i=1;i<=n;++i){
		f[i]=i;
		for(re int j=1;j<=n;++j){
			scanf("%d",&m);
			if(i<j){
				tot[++l].x=i;
				tot[l].y=j;
				tot[l].num=m;
			}
		}
	}
	std::sort(tot+1,tot+l+1,cmp);
	m=1;
	for(re int i=1;i<=l;++i){
		if(find(tot[i].x)!=find(tot[i].y)){
			ans+=tot[i].num;
			f[find(tot[i].x)]=tot[i].y;
			++m;
			if(n==m)
			break;
		}
	}
	printf("%d",ans);
	return 0;
}

标程:

#include<bits/stdc++.h>

using namespace std;

struct node{
	int x,y,w;
}a[200002];

int f[200002];

bool cmp(node xx,node yy){//结构体排序
	return xx.w<yy.w;
}

int find(int x){
//并查集说白了就是找父结点的过程,同一个父节点即同一个区间
	if(x==f[x]) return x;
	f[x]=find(f[x]);
	return f[x];
}

int main(){
	int n,k,m=0;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		f[i]=i;
		for(int j=1;j<=n;j++)
		{
			scanf("%d",&k);
			if(j>i){
            //读入时加一个判断就可以了,不需要读那么多
				m++;
				a[m].x=i;a[m].y=j;a[m].w=k;	
			}	
		}
	}
	sort(a+1,a+m+1,cmp);//排序
	int ans=0,p=1;
	for(int i=1;i<=m;i++){
		if(find(a[i].x)!=find(a[i].y)){
        //如果不在一个集合
			ans+=a[i].w;
			f[find(a[i].x)]=a[i].y;
            //合并两个节点
			p++;
			if(p==n) break;	
		}
	}
	cout<<ans;
	return 0;
}
2022/2/26 19:44
加载中...