题解 | #继续畅通工程#

继续畅通工程

https://www.nowcoder.com/practice/16212f7d46e44174b5505997ea998538

令weight为0即可
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

struct Edge {
	int x;
	int y;
	int weight;
	bool finish;  //表示完成的状态
};

#define N 1000 //元素的上限个数
int father[N];
int height[N];

bool compare(Edge lhs, Edge rhs) {
	return lhs.weight < rhs.weight;
}


void Init(int n) {
	for (int i = 1; i <= n; i++) {
		father[i] = i;
		height[i] = 1;
	}
}

int Find(int x) {
	if (x != father[x]) {
		father[x] = Find(father[x]);
	}
	return father[x];

}

void Union(int x, int y,int weight,int & total) {
	x = Find(x); y = Find(y);
	if (x != y) {  //判断x与y是否属于同一集合
		total += weight;  
	}
	if (height[x] < height[y])
		father[x] = y;
	else if (height[x] > height[y])
		father[y] = x;
	else {
		father[y] = x;
		++height[x];
	}
}

int main() {
	int n;
	while (cin >> n) {
		if (n == 0) break;
		Init(n);
		vector<Edge>  vec;
		Edge temp;
		for (int i = 0; i < n * (n - 1)/2; i++) {
			cin >> temp.x >> temp.y >> temp.weight >> temp.finish;
			if (temp.finish == true) 
				temp.weight = 0;

			vec.push_back(temp);
		}
		sort(vec.begin(), vec.end(), compare);  //将集合进行排序
		int total = 0;
		for (int i = 0; i < n * (n - 1)/2; i++) {
			 Union(vec[i].x, vec[i].y, vec[i].weight, total);
		}
		cout << total << endl;

	}
}

全部评论

相关推荐

来,说点可能被同行“骂”的大实话。🙊当初接数字马力Offer时,朋友都说:“蚂蚁的“内包”公司?你想清楚啊!”但入职快一年后的今天,我反而对他有了不一样的看法!🔹&nbsp;是偏见?还是信息差!之前没入职之前外面都在说什么岗位低人一等这类。实际上:这种情况不可至否,不能保证每个团队都是其乐融融。但我在的部门以及我了解的周边同事都还是十分好相处的~和蚂蚁师兄师姐之间也经常开一些小玩笑。总之:身份是蚂蚁公司给的,地位是自己挣的(一个傲娇女孩的自述)。🔹&nbsp;待遇?玩的就是真实!试用期工资全额发!六点下班跑得快(早9晚6或者早10晚7,动态打卡),公积金顶格交。别听那些画饼的,到手的钱和下班的时间才是真的(都是牛马何必难为牛马)。🔹&nbsp;能不能学到技术?来了就“后悔”!我们拥有权限直通蚂蚁知识库,技术栈多到学不完。说“学不到东西”的人,来了可能后悔——后悔来晚了(哈哈哈哈,可以不学但是不能没有)!💥&nbsp;内推地址:https://app.mokahr.com/su/ueoyhg❗我的内推码:NTA6Nvs走我的内推,可以直达业务部门,面试流程更快速,进度可查!今天新放HC,之前挂过也能再战!秋招已经正式开始啦~机会就摆在这,敢不敢来试一试呢?(和我一样,做个勇敢的女孩)
下午吃泡馍:数字马力的薪资一般哇,5年经验的java/测试就给人一万出头,而且刚入职第三天就让人出差,而且是出半年
帮你内推|数字马力 校招
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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