Prim算法堆优化

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <ctype.h>
#include <math.h>
#include <iostream>
#include <string>
#include <stack>
#include <vector>
#include <set>
#include <queue>
#include <algorithm>
#define LL long long int
#define N 100000+10 //最大节点数
#define M 1000000+10 //最大的边数
#define MOD 142857
//N<=10^5, M<=10^6 using namespace std;
int n, m;
struct node
{
int v, w;
bool operator<(const node &dd)const{
return w>dd.w;
} //权值小的优先
};
vector<node>q[N];
bool vis[N];
//堆优化的prim算法
LL ans; void queue_prim()
{
//以节点1为起点进行扩展安全边 生成最小树
priority_queue<node>que;
while(!que.empty())
que.pop(); //初始化清空优先队列 维护一个小根堆
//这样每次找安全边的速度就提高了
ans = ;
memset(vis, false, sizeof(vis));
for(int i=; i<q[].size(); i++){
que.push(q[][i]); //将起点的所有连接边全部加入队列中来
}
vis[]=true;
int edge=n-;//边数
node cur;
while(edge--)
{
cur = que.top();
que.pop();//这个地方需要注意一下
//并不是每个从优先队列取出来的边都是可以加到生成树上去的 if(vis[cur.v]==true){
while(vis[cur.v]){
cur=que.top(); que.pop();
}
}
ans = ans+cur.w; //printf("%d-- ", cur.w );
vis[cur.v]=true; //加入生成树的该点将被标记访问
for(int i=; i<q[cur.v].size(); i++){
if(vis[ q[cur.v][i].v ]==false) //当前加入生成树的点可以扩充出的边指向的节点
que.push(q[cur.v][i]);//如果没有被访问才会加入到队列当中来
}
}
} int main()
{
scanf("%d %d", &n, &m);
int i, j;
int u, v, w;
node cur;
for(i=; i<=n; i++)
q[i].clear(); for(i=; i<m; i++)
{
scanf("%d %d %d", &u, &v, &w);
cur.v=v; cur.w=w;
q[u].push_back(cur);
cur.v=u;
q[v].push_back(cur); //建立双向边
}
queue_prim();
printf("%lld\n", ans );
return ;
}
上一篇:dijkstra(最短路)和Prim(最小生成树)下的堆优化


下一篇:P3366 【模板】最小生成树(堆优化prim)