最小生成树 （prim 模板）

题目链接

存个模板！！

prim：用小根堆优化的时间复杂度是（V + E）logV，时间复杂度跟dijkstra一样，因为实现思路基本一样。

该算法巧妙的利用了，最小生成树的切分性质：给定任意切分，横切边最小的边必然属于最小生成树（横切边是指一个端点在最小生成树中，另一个不在）。

update：其实感觉没太必要，这个时间复杂度直接kruskal就好了，不需要堆优化的最小生成树。

朴素prime：复杂度o（n*n）

#include <bits/stdc++.h>
#define ll long long
#define pi pair<int,int>
#define mk make_pair
#define pb push_back
using namespace std;
const int maxn = 5005;
int a[maxn][maxn], d[maxn], vis[maxn], n;
void primes()
{
	memset(vis, 0, sizeof(vis));
	memset(d, 0x3f, sizeof(d));
	d[1] = 0;
	for(int i = 1; i < n; i++)
	{
		int x = 0;
		for(int j = 1; j <= n; j++)
			if(!vis[j] && (x == 0 || d[j] < d[x]))x = j;
		vis[x] = 1;
		for(int j = 1; j <= n; j++)
			if(!vis[j])d[j] = min(d[j] , a[x][j]);
	}
	
	int ans = 0;
	for(int i = 2; i <= n; i++)ans += d[i];
	printf("%d\n",ans);
	return ;
}

int main()
{
	int m;
	cin >> n >> m;
	memset(a, 0x3f, sizeof(a));
	for(int i = 1; i <= n; i++)a[i][i] = 0;
	for(int i = 1; i <= m; i++)
	{
		int u, v, w;
		cin >> u >> v >> w;
		a[u][v] = a[v][u] = min(a[u][v] , w);
	}
	
	primes();
	return 0;
}

 

堆优化的最小生成树：

#include <bits/stdc++.h>
#define pb push_back
#define ll long long
#define pi pair<int,int>
#define mk make_pair
using namespace std;
const int maxn = 105;
int n;
vector<pi>G[maxn];
bool vis[maxn];

ll prim()
{
	memset(vis,0,sizeof(vis));
	ll res = 0;
	priority_queue<pi>q;
	q.push(mk(0,1));
	while(!q.empty())
	{
		int u = q.top().second;
		int w = -q.top().first;
		q.pop();
		if(vis[u])continue;
		vis[u] = 1;
		res += w;
		for(int i=0;i<G[u].size();i++)
			q.push(mk(-G[u][i].second,G[u][i].first));
	}
	return res;
}
int main()
{
	while(scanf("%d",&n) == 1&&n)
	{
		int m = n*(n-1)/2;
		for(int i=1;i<=m;i++)
		{
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			G[u].pb(mk(v,w));
			G[v].pb(mk(u,w));
		}
		printf("%lld\n",prim());
		for(int i=1;i<=n;i++)G[i].clear();
	}
}