Prim

最新推荐文章于 2024-08-12 19:29:12 发布

J_Sure

最新推荐文章于 2024-08-12 19:29:12 发布

阅读量874

点赞数

分类专栏： Prim 模板文章标签： acm c++ oj

本文链接：https://blog.csdn.net/j_sure/article/details/38445179

版权

模板同时被 2 个专栏收录

32 篇文章 0 订阅

订阅专栏

Prim

8 篇文章 0 订阅

订阅专栏

输入：

N（顶点） M（边）

u v w（权值）

输出：

建图过程、最小生成树的权值

Sample Input:

7 9
1 2 28
1 6 10
2 3 16
2 7 14
3 4 12
4 5 22
4 7 18
5 6 25
5 7 24

Sample Output:

1 6 10
6 5 25
5 4 22
4 3 12
3 2 16
2 7 14
weight of MST is 99

————————————————————集训8.8的分割线————————————————————

思路：Prim模板

代码如下：

/*
ID: j.sure.1
PROG:
LANG: C++
*/
/****************************************/
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <stack>
#include <queue>
#include <vector>
#include <map>
#include <string>
#include <iostream>
#define INF 2e9
using namespace std;
/****************************************/
const int N = 105;
int n, m, sum;
int mat[N][N], dis[N], near[N];

int prim(int st)
{
	int sum = 0;
	for(int i = 1; i <= n; i++) {
		dis[i] = mat[st][i];
		near[i] = st;
	}
	near[st] = -1;
	for(int i = 1; i <= n; i++) {
		int k = -1, mini = INF;
		for(int j = 1; j <= n; j++) if(near[j] != -1) {
			if(dis[j] < mini) {
				mini = dis[j];
				k = j;
			}
		}
        if(k == -1) return 0;
        near[k] = -1;
        sum	+= mini;
        for(int j = 1; j <= n; j++) if(near[j] != -1) {
            if(dis[j] > mat[k][j]) {
                near[j] = k;
                dis[j] = mat[k][j];
            }
        }
	}
	printf("weight of MST is %d\n", sum);
    return sum;
}

int main()
{
#ifdef J_Sure
	freopen("test.in", "r", stdin);
//	freopen(".out", "w", stdout);
#endif
	scanf("%d%d", &n, &m);
	for(int i = 0; i < m; i++) {
		int a, b;
		scanf("%d%d", &a, &b);
		scanf("%d", &mat[a][b]);
		mat[b][a] = mat[a][b];
	}

	for(int i = 1; i <= n; i++) {
		for(int j = 1; j <= n; j++) {
			if(i == j)
				mat[i][j] = 0;
			else if(mat[i][j] == 0)
				mat[i][j] = INF;
		}
	}
	prim(1);
	return 0;
}