题解 | #还是畅通工程#

还是畅通工程

http://www.nowcoder.com/practice/d6bd75dbb36e410995f8673a6a2e2229

#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std; 

const int MAXN = 1000;

struct Edge{    //定义边结构体
	int from;
	int to;
	int length;
};

Edge edge[MAXN * MAXN];   //边的数量是定点数量的平方

int father[MAXN];    //用以表示每个节点的父节点是什么样的

int height[MAXN];   

void Initial(int n){    //n个节点
	for(int i = 0; i < n; ++i){    //初始状态下,每个节点的父节点是其自身
		father[i] = i;
		height[i] = 0;
	}
	return;
}

int Find(int x){
	if(x != father[x]){  //表示当前节点不是根节点,需要继续向上查找
		father[x] = Find(father[x]);    //将路径压缩,可以提高查找效率
	}
	return father[x];
}

void Union(int x,int y){  //合并,将一棵树作为另一棵树的子树
	x = Find(x);   //找到x的根节点
	y = Find(y);    //找到y的根节点
	if(x != y){    //说明这两个节点本身不属于同一个集合,需合并
		if(height[x] < height[y]){   //将树高较高的作为树高较低的子树进行合并
			father[x] = y;    
		}else if(height[x] > height[y]){
			father[y] = x;    
		}else{   //高度相同时,可任意连接
			father[y] = x;   
			height[x]++;   //树的高度递增1
		}
	}
}

bool Compare(Edge x,Edge y){
	return x.length < y.length;
}

int Kruskal(int n,int edgeNum){
	Initial(n);   //先初始化
	sort(edge,edge + edgeNum,Compare);  //按照边的权值升序排序
	int sum = 0;   //统计路径之和
	for(int i = 0; i < edgeNum; ++i){
		Edge current = edge[i];
		if(Find(current.from) != Find(current.to)){   //两个定点不属于同一个集合,就进行合并
			Union(current.from,current.to);
			sum += current.length;
		}
	}
	return sum;
}



int main(){
	int n;
	while(scanf("%d",&n) != EOF){
		if(n == 0){
			break;
		}
		int edgeNum = n * (n - 1) / 2;   //n个顶点,n * (n - 1) / 2条边
		for(int i = 0; i < edgeNum; ++i){    //输入边的信息
			scanf("%d%d%d",&edge[i].from,&edge[i].to,&edge[i].length);
		}
		int answer = Kruskal(n,edgeNum);
		printf("%d\n",answer);
	}
	return 0;
}
全部评论
感觉有问题,我试了多组数据,每组数前面几个测试样例没问题,后面就出现错误了
点赞 回复 分享
发布于 01-25 18:59 广西

相关推荐

点赞 评论 收藏
分享
2 收藏 评论
分享
牛客网
牛客企业服务