【洛谷】P2820最小生成树经典例题代码及详解

题目描述

通过的代码

#include<iostream>
#include<queue>
using namespace std;
struct node{
	int index;
	int dis;
	node(int i,int d):index(i),dis(d)
	{}
};
struct cmp{
	bool operator()(node a,node b)
	{
		return a.dis>b.dis;
	}
};
int vis[110]; 
int dist[110][110];
int main()
{
	int count=0;
	int TreeLength=0;
	int TotalLength=0;
	priority_queue<node,vector<node>,cmp>no;
	int n,k;
	int start,end,m;
	int CurrentIndex;
	int ans;
	cin>>n>>k;
	while(k--)
	{
	    cin>>start>>end>>m;
		dist[start][end]=m;
		dist[end][start]=m;	
		TotalLength+=m;
	}
	node PerNode(1,0);
	no.push(PerNode);
	while(!no.empty())
	{
		node CurrentNode=no.top();
		no.pop();
		CurrentIndex=CurrentNode.index;
		if(vis[CurrentIndex]==1)
		{
			continue;
		}
		vis[CurrentIndex]=1;
		TreeLength+=CurrentNode.dis;
		count++;
		if(count==n)
		break;
		for(int i=0;i<=n;i++)
		{
			if(i==CurrentIndex)
			continue;
			if(vis[i]==0&&dist[CurrentIndex][i]!=0)
			{
				node nod(i,dist[CurrentIndex][i]);
				no.push(nod);
			}
		}
	}
	ans=TotalLength-TreeLength;
	cout<<ans<<endl;
	return 0;
}

解题思路

这是一道比较经典的最小生成树的题目，我是用prime解决的，因为之前没有用过prime。prime算法其实类似于bfs。因为bfs一般是用队列解决，所以prime也是，但是因为prime每次都需要将权值最小的一条边加到树里面，所以用简单的队列还是解决不了，必须要用优先队列，优先队列的特点就是可以每次对队列中的值进行排序，每次top队首的结点都是优先级最高的。这个题目的思路，简而言之，就是在while循环中，每次得到队首结点top，并且删除该元素，此时队首对应的结点就被加入到了生成树中。此时需要将该结点的所有还未被加入到生成树中的邻接节点都加入到优先队列中。这些结点可能之前已经被加入过了，但是此时，它们的dis距离值（可以对应代码中struct结构体中的dis）发生了变化，所以还需要重新push到优先队列中。因为优先队列不能对队列中的值进行去重等操作，所以在队列中，一个结点可能被push了多次（每次push的时候都有一个不同的dis值）。为了避免之前已经被加到生成树中的结点再一次被重复加到树中，需要建立一个vis数组，保存每一个结点是否已经被加入到树中。
需要注意一点，优先队列的比较函数跟我们正常的不一样，上面代码中的cmp最后是形成了一个小顶堆，实现从小到大的排序。这里很重要！！