51Nod 1212 无向图最小生成树(模板)

N个点M条边的无向连通图,每条边有一个权值,求该图的最小生成树。
Input
第1行:2个数N,M中间用空格分隔,N为点的数量,M为边的数量。(2 <= N <= 1000, 1 <= M <= 50000)
第2 - M + 1行:每行3个数S E W,分别表示M条边的2个顶点及权值。(1 <= S, E <= N,1 <= W <= 10000)
Output
输出最小生成树的所有边的权值之和。
Input示例
9 14
1 2 4
2 3 8
3 4 7
4 5 9
5 6 10
6 7 2
7 8 1
8 9 7
2 8 11
3 9 2
7 9 6
3 6 4
4 6 14
1 8 8
Output示例
37

第一种:vector数组实现。

#include<bits/stdc++.h>
using namespace std;
struct  stu{
    int x;
    int y;
    int cost;
}; 
int f[105];
int find(int x){
	if(x!=f[x])
	f[x]=find(f[x]); 
	return f[x];
} 
bool unite(int x,int y){               //并查集
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	{f[fx]=fy;
	 return true;                 //给出提示,假如合并了返回真值
	}
	return false;
}
bool cmp(stu a,stu b){
	return a.cost<b.cost;                //权值从小到大排序
}
int main(){
	int n,m;
	stu r; 
	vector<stu> v;
	while(~scanf("%d%d",&n,&m)&&n){
		v.clear();
		for(int i=1;i<=n;i++){
			 f[i]=i;
		}
		for(int i=1;i<=m;i++){
			scanf("%d%d%d",&r.x,&r.y,&r.cost);
			v.push_back(r);
		}
		sort(v.begin(),v.end(),cmp);
		int sum=0,cnt=0;
		for(int i=0;i<v.size();i++){
			if(unite(v[i].x,v[i].y)){
				sum+=v[i].cost;
				cnt++; 
			}
			if(cnt==n-1)
			break;
		}
		printf("%d\n",sum);
	}
	return 0;
}

 

2.也可以直接结构体数组,可能这个内存大点.

 

#include<bits/stdc++.h>
using namespace std;
const int maxn=1010;
const int maxm=50010;
int f[maxn];
struct edge{
  int u,v;
  int len;
}e[maxm];
bool cmp(edge a,edge b){
    return a.len<b.len;
}
void init(int n){//初始化并查集
    for(int i=1;i<=n;i++)
	f[i]=i;
}
int find(int x){
    if(x!=f[x])
    f[x]=find(f[x]);
    return f[x];
}
int unite(int x,int y){  //合并两个集合(如果x,y在同一集合,返回0,否则返回1)
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy){
		f[fx]=fy;
		return 1;
	}
	return 0;
}
int main(){
	int m,n;
	while(~scanf("%d%d",&n,&m)){
		init(n);
		for(int i=1;i<=m;i++){
		scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].len);
		}
		sort(e+1,e+m+1,cmp);      //注意!这里千万不能错,注意排序范围
		int sum=0;
		for(int i=1;i<=m;i++){
		 {
		 sum+=e[i].len*unite(e[i].u,e[i].v); //不必再加个ant来优化循环了
		 }                                   //通过unite(i,j)的值决定加不加
 		}                                    //本质上意思和用if语句那个一样
		printf("%d\n",sum);
	}
	return 0;
}


第二种在最后计算sum的步骤里,可能语句稍微简洁点,但是数据量大的情况下,还是第一种占用内存少.

 

 

 

 

 

全部评论

相关推荐

不愿透露姓名的神秘牛友
2025-12-17 16:48
今天九点半到公司,我跟往常一样先扫了眼电脑,屁活儿没有。寻思着没事干,就去蹲了个厕所,回来摸出手机刷了会儿。结果老板刚好路过,拍了我一下说上班别玩手机,我吓得赶紧揣兜里。也就过了四十分钟吧,我的直属领导把我叫到小隔间,上来就给我一句:“你玩手机这事儿把老板惹毛了,说白了,你可以重新找工作了,等下&nbsp;HR&nbsp;会来跟你谈。”&nbsp;我当时脑子直接宕机,一句话都没憋出来。后面&nbsp;HR&nbsp;找我谈话,直属领导也在旁边。HR&nbsp;说我这毛病不是一次两次了,属于屡教不改,不光上班玩手机,还用公司电脑看论文、弄学校的事儿。我当时人都傻了,上班摸鱼是不对,可我都是闲得发慌的时候才摸啊!而且玩手机这事儿,从来没人跟我说过后果这么严重,更没人告诉我在公司学个习也算犯错!连一次口头提醒都没有,哪儿来的屡教不改啊?更让我膈应的是,昨天部门刚开了会,说四个实习生里留一个转正,让大家好好表现。结果今天我就因为玩手机被开了。但搞笑的是,开会前直属领导就把我叫去小会议室,明明白白告诉我:“转正这事儿你就别想了,你的学历达不到我们部门要求,当初招你进来也没打算给你这个机会。”合着我没入贵厂的眼是吧?可我都已经被排除在转正名单外了,摸个鱼至于直接把我开了吗?真的太离谱了!
rush$0522:转正名单没进,大概率本来就没打算留你
摸鱼被leader发现了...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务