代码随想录算法训练营第五十七天 | 图论part07

53. 寻宝 prim算法

prim算法

cpp 复制代码
#include <iostream>
#include <vector>
#include <fstream>
#include <climits>

using namespace std;

int main() {
	int v, e;
	int v1, v2, val;
	ifstream infile("input.txt");
	cin >> v >> e;
	vector<vector<int>> graph(v + 1, vector<int>(v + 1, INT_MAX));
	while (e--) {
		cin >> v1 >> v2 >> val;
		graph[v1][v2] = val;
		graph[v2][v1] = val;
	}
	
	/*for (int i = 0; i < v + 1; ++i) {
		for (int j = 0; j < v + 1; ++j) {
			cout << "v1: " << i << " v2: " << j << " val: " << graph[i][j] << endl;
		}
	}*/

	vector<bool> isInTree(v + 1, false);
	vector<int> minDest(v + 1, INT_MAX);

	// 找到第一个顶点
	// 将它放入树中
	// 更新minDest
	isInTree[1] = true;
	for (int j = 1; j < v + 1; ++j) {
		if (!isInTree[j] && graph[1][j] < minDest[j])
			minDest[j] = graph[1][j];
	}

	for (int i = 2; i < v; ++i) { // 只需要循环v-1次
		int curMin = 0;
		int minVal = INT_MAX;
		for (int j = 1; j < v + 1; ++j) {
			if (!isInTree[j] && minDest[j] < minVal) {
				minVal = minDest[j];
				curMin = j;
			}
		}

		isInTree[curMin] = true;

		for (int j = 1; j < v + 1; ++j) {
			if (!isInTree[j] && graph[curMin][j] < minDest[j])
				minDest[j] = graph[curMin][j];
		}
	}
	int result = 0;
	for (int i = 2; i <= v; ++i) {
		result += minDest[i];
	}
	cout << result << endl;
	return 0;
}

kruskal算法

算法思路是按边的val进行从小到大的排序,在不出现环的情况下,将这个边作为最小完全树的一部分。

cpp 复制代码
#include <iostream>
#include <vector>
#include <fstream>
#include <algorithm>

using namespace std;

struct Edge {
	int v1, v2, val;
};

void init(vector<int>& father) {
	for (int i = 0; i < father.size(); ++i) {
		father[i] = i;
	}
}

int find(vector<int>& father, int u) {
	if (father[u] == u) return u;
	return father[u] = find(father, father[u]);
}

bool same(vector<int>& father, int u, int v) {
	u = find(father, u);
	v = find(father, v);
	if (u == v) return true;
	else return false;
}

void join(vector<int>& father, int u, int v) {
	u = find(father, u);
	v = find(father, v);
	if (u == v) return;
	else father[v] = u;
}

int main() {
	int v, e;
	int v1, v2, val;
	ifstream infile("input.txt");
	cin >> v >> e;
	vector<Edge> edges;
	while (e--) {
		cin >> v1 >> v2 >> val;
		edges.emplace_back(Edge{v1, v2, val});
	}
	
	/*for (int i = 0; i < v + 1; ++i) {
		for (int j = 0; j < v + 1; ++j) {
			cout << "v1: " << i << " v2: " << j << " val: " << graph[i][j] << endl;
		}
	}*/
	vector<int> father(v + 1, 0);
	sort(edges.begin(), edges.end(), [](const Edge &a, const Edge &b) {
		return a.val < b.val;
		});

	init(father);

	int result = 0;

	for (Edge edge : edges) {
		if (!same(father, edge.v1, edge.v2)) {
			join(father, edge.v1, edge.v2);
			result += edge.val;
		}
	}
	cout << result << endl;
	return 0;
}
相关推荐
机器学习之心7 小时前
多目标鲸鱼优化算法(NSWOA),含46种测试函数和9个评价指标,MATLAB实现
算法·matlab·多目标鲸鱼优化算法·46种测试函数·9个评价指标
max5006008 小时前
基于Meta Llama的二语习得学习者行为预测计算模型
人工智能·算法·机器学习·分类·数据挖掘·llama
王哥儿聊AI9 小时前
Lynx:新一代个性化视频生成模型,单图即可生成视频,重新定义身份一致性与视觉质量
人工智能·算法·安全·机器学习·音视频·软件工程
手握风云-11 小时前
优选算法的寻踪契合:字符串专题
算法
闭着眼睛学算法11 小时前
【华为OD机考正在更新】2025年双机位A卷真题【完全原创题解 | 详细考点分类 | 不断更新题目 | 六种主流语言Py+Java+Cpp+C+Js+Go】
java·c语言·javascript·c++·python·算法·华为od
IT古董11 小时前
【第五章:计算机视觉-项目实战之目标检测实战】2.目标检测实战:中国交通标志检测-(2)中国交通标志检测数据格式转化与读取
算法·目标检测·计算机视觉
MobotStone11 小时前
LLM 采样入门到进阶:理解与实践 Top-K、Top-P、温度控制
算法
杨小码不BUG12 小时前
CSP-J/S初赛知识点精讲-图论
c++·算法·图论··编码·csp-j/s初赛
LeaderSheepH13 小时前
常见的排序算法
数据结构·算法·排序算法
周杰伦_Jay14 小时前
【图文详解】强化学习核心框架、数学基础、分类、应用场景
人工智能·科技·算法·机器学习·计算机视觉·分类·数据挖掘