算法训练营|图论第7天 prim算法 kruskal算法

题目:prim算法

题目链接:

53. 寻宝(第七期模拟笔试) (kamacoder.com)

代码:

cpp 复制代码
#include<bits/stdc++.h>
#include<unordered_map>
#include<unordered_set>
using namespace std;
int main() {
	int v, e;
	int x, y, k;
	cin >> v >> e;
	vector<vector<int>>grid(v + 1, vector<int>(v + 1, 10001));
	while (e--) {
		cin >> x >> y >> k;
		grid[x][y] = k;
		grid[y][x] = k;
	}
	vector<int>minEdges(v + 1, 10001);
	vector<bool>Intree(v + 1, false);
	for (int i = 1; i < v; i++) {
		int cur = -1;
		int MinVal = INT_MAX;
		for (int j = 1; j <= v; j++) {
			if (!Intree[j] && minEdges[j] < MinVal) {
				cur = j;
				MinVal = minEdges[j];
			}
		}
		Intree[cur] = true;
		for (int j = 1; j <= v; j++) {
			if (!Intree[j] && grid[cur][j] < minEdges[j]) {
				minEdges[j] = grid[cur][j];
			}
		}
	}
	int result = 0;
	for (int i = 2; i <= v; i++) {
		result += minEdges[i];
	}
	cout << result << endl;
}

题目:kruskal算法

题目链接:

53. 寻宝(第七期模拟笔试) (kamacoder.com)

代码:

cpp 复制代码
#include<bits/stdc++.h>
#include<unordered_map>
#include<unordered_set>
using namespace std;

struct Edge {
	int l, r, val;
};
int n = 10001;
vector<int>father(n, -1);
void init() {
	for (int i = 0; i < n; i++) {
		father[i] = i;
	}
}
int find(int u) {
	if (u == father[u]) return u;
	else {
		return father[u] = find(father[u]);
	}
}
void join(int u, int v) {
	u = find(u);
	v = find(v);
	if (u == v) return;
	father[v] = u;

}
static bool cmp(const Edge& a, const Edge& b) {
	return a.val < b.val;
}

int main() {
	int v, e;
	int x, y, k;
	cin >> v >> e;
	vector<Edge>edges;
	while (e--) {
		cin >> x >> y >> k;
		edges.push_back({ x,y,k });
	}
	sort(edges.begin(), edges.end(), cmp);
	vector<Edge>result;
	init();
	int result_val = 0;
	for (Edge edge : edges) {
		int x = find(edge.l);
		int y = find(edge.r);
		if (x != y) {
			result.push_back(edge);
			result_val += edge.val;
			join(x, y);
		}
	}
	
	cout << result_val << endl;
}
相关推荐
xiaoshiguang315 分钟前
LeetCode:222.完全二叉树节点的数量
算法·leetcode
爱吃西瓜的小菜鸡16 分钟前
【C语言】判断回文
c语言·学习·算法
别NULL18 分钟前
机试题——疯长的草
数据结构·c++·算法
TT哇23 分钟前
*【每日一题 提高题】[蓝桥杯 2022 国 A] 选素数
java·算法·蓝桥杯
ZSYP-S2 小时前
Day 15:Spring 框架基础
java·开发语言·数据结构·后端·spring
yuanbenshidiaos2 小时前
C++----------函数的调用机制
java·c++·算法
唐叔在学习2 小时前
【唐叔学算法】第21天:超越比较-计数排序、桶排序与基数排序的Java实践及性能剖析
数据结构·算法·排序算法
ALISHENGYA2 小时前
全国青少年信息学奥林匹克竞赛(信奥赛)备考实战之分支结构(switch语句)
数据结构·算法
chengooooooo2 小时前
代码随想录训练营第二十七天| 贪心理论基础 455.分发饼干 376. 摆动序列 53. 最大子序和
算法·leetcode·职场和发展
jackiendsc2 小时前
Java的垃圾回收机制介绍、工作原理、算法及分析调优
java·开发语言·算法