最小生成树

 Prim算法

#include <iostream>

using namespace std;

// http://t.csdn.cn/rFme4




/*

朴素普利姆算法
适用：稠密图


还有堆优化版prim算法（类似与dijkstra算法），用堆维护dist数组
*/

const int N = 510;

// 点到集合的距离
int dist[N];

// 稠密图存储用临界矩阵
int g[N][N];

// 存储该点是否在生成树中
bool st[N];

int n, m;

// 如果生成树存在，返回生成树所有边权重之和；如果不存在返回无穷大(0x3f3f3f3f)
int prim()
{
	// 初始化距离为无穷大
	memset(dist, 0x3f, sizeof dist);
	dist[1] = 0;

	// 记录生成树中所有边权重的和
	int res = 0;

	// 每次循环将一个点加入生成树中
	for (int i = 0; i < n; i++)
	{
		// 0为无效点，且0为合法空间，可用为比较点
		int t = 0;

		// 在所有未进入生成树的点中找出距离树最近的点（dist最小）
		for (int j = 1; j <= n; j++)
		{
			if (!st[j] && dist[j] < dist[t])
			{
				t = j;
			}
		}

		// 如果距离进入树最近的点的距离也为无穷大，说明该无向图无法生成最小生成树
		if (dist[t] == 0x3f3f3f3f)
		{
			return 0x3f3f3f3f;
		}
		else
		{
			// 否则进入生成树
			st[t] = true;

			// 更新权重（先更新权重再更新其他点，不然可能会受自环影响，自环是不应该加入生成树中的）
			res += dist[t];

			// 用当前点更新其他点到树的距离
			for (int j = 1; j <= n; j++)
			{
				dist[j] = min(dist[j], g[t][j]);
			}
		}
	}

	return res;
}

int main()
{
	// 初始化邻接矩阵
	memset(g, 0x3f, sizeof g);

	cin >> n >> m;

	for (int i = 0; i < m; i++)
	{
		int a, b, c;
		cin >> a >> b >> c;

		g[a][b] = g[b][a] = min(g[a][b], c);
	}

	int t = prim();

	if (t == 0x3f3f3f3f)
	{
		cout << "impossible" << endl;
	}
	else
	{
		cout << t << endl;
	}

	return 0;
}

Kruskal算法

#include <iostream>
#include <algorithm>
using namespace std;

/*

克鲁斯卡尔(kruskal)算法
适用：稀疏图

*/

const int N = 100010;
const int M = 200020;

int n, m;
int p[N]; // 并查集父节点数组

// a与b的边权重为c
struct Edge
{
	int a, b, c;

	// 重载小于号
	bool operator< (const Edge& e)const
	{
		return c < e.c;
	}
}edges[M];


// 并查集核心操作（找祖宗节点）
int find(int x)
{
	if (p[x] != x)
	{
		p[x] = find(p[x]);
	}

	return p[x];
}

// 将两个集合并在一起
void merge(int a, int b)
{
	p[find(a)] = find(b);
}



// 返回最小生成树中边的权重之和，如果最小生成树不存在则返回无穷大(0x3f3f3f3f)
int kruskal()
{
	// 当前生成树中边的个数
	int cnt = 0;

	// 生成树中边的权重之和
	int res = 0;

	// 按边的权重对边进行排序
	sort(edges, edges + m);

	// 从小到大遍历每一条边
	for (int i = 0; i < m; i++)
	{
		int a = edges[i].a;
		int b = edges[i].b;
		int c = edges[i].c;

		// 如果两个点不连通
		if (find(a) != find(b))
		{
			merge(a, b);
			res += c;
			cnt++;
		}
	}

	// 最小生成树中边的个数一定为点的个数-1，如果不是说明最小生成树不存在
	if (cnt < n - 1)
	{
		return 0x3f3f3f3f;
	}
	else
	{
		return res;
	}
}

int main()
{
	cin >> n >> m;

	// 初始化并查集
	for (int i = 0; i <= n; i++)
	{
		p[i] = i;
	}

	for (int i = 0; i < m; i++)
	{
		int a, b, c;
		cin >> a >> b >> c;
		edges[i] = { a,b,c };
	}

	int t = kruskal();

	if (t == 0x3f3f3f3f)
	{
		cout << "impossible" << endl;
	}
	else
	{
		cout << t << endl;
	}

	return 0;
}