USACO Agri-Net, MST, Prim, Kruskal

最新推荐文章于 2024-02-04 14:07:20 发布

原创最新推荐文章于 2024-02-04 14:07:20 发布 · 390 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #算法

USACO 专栏收录该内容

43 篇文章

订阅专栏

本文介绍使用Prim算法解决最小生成树问题，并提供了一个具体的C++实现案例。该案例通过读取输入文件来构建图，并逐步展示如何选择边来形成最小生成树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最基本的最小生成树，可以用Prim和Kruskal

Prim算法的解答，参考http://blog.youkuaiyun.com/niushuai666/article/details/6689295

/* 
ID: wangxin12 
PROG: agrinet
LANG: C++ 
*/ 
#include <iostream>
#include <vector>
#include <fstream>
#include <queue>
using namespace std;
#define MAX 100

long map[MAX + 1][MAX + 1];
long lowcost[MAX + 1];
bool visited[MAX + 1];
int nodeNum, sum = 0;

void prim();

int main() {
	ifstream fin("agrinet.in");
	ofstream fout("agrinet.out");
	int i, j;
	//read data
	fin>>nodeNum;
	for(i = 1; i <= nodeNum; i++) {
		for(j = 1; j <= nodeNum; j++) {
			fin>>map[i][j];
		}
	}
	//body prim
	prim();
	fout<<sum<<endl;

	return 0;
}

void prim() {
	long temp, k;
	int i, j;  //循环用
	
	//init,选择prim开始的node，此处选1
	visited[1] = true;
	for(i = 1; i <= nodeNum; i++) {
		lowcost[i] = map[1][i];
	}
	//
	for(i = 1; i <= nodeNum; i++) {
		temp = 999999999;
		for(j = 1; j <= nodeNum; j++) {
			if( !visited[j] && temp > lowcost[j]) {
				temp = lowcost[j];
				k = j;
			}
		}
		if(temp == 999999999)
			break;
		visited[k] = true;
		sum += lowcost[k];
		for(int t = 1; t <= nodeNum; t++) {
			if( !visited[t] && lowcost[t] > map[k][t])
				lowcost[t] = map[k][t];
		}
	}
}