hihoCoder-1109-堆优化的Prim

优先队列是由堆组成的,所以当我们使用优先队列对Prim进行优化时,就把这种优化叫做堆优化。
它的算法核心思想就是每次向后找边,每个pair存的都是下一个点,以及边权。我们对于已经走过的点就避开,这样就不会形成环。
因为我们已知的点是标记过的,我们只要下一个点不走已知的点肯定不会形成环路。因为第一个点已经标记过,所以我们之后只用做n-1次,找n-1个点就可以了,顺便找了n-1条边,这时候一颗最小生成树就建成了。
对于那pair我们把权值放在第一位,然后调用系统的升序就可以了,就不用自己写排序的函数了。因为优先队列对pair进行排序的时候默认是按照第一个数字进行排序的。

#include <iostream>
#include <cstring>
#include <vector>
#include <queue>
using namespace std;
const int maxn = 100010;
typedef pair<int, int> p;
vector<p> vec[maxn];
int vis[maxn];

int main()
{
    int n, m, u, v, val;
    scanf("%d%d", &n, &m);
    for (int i = 0; i < m;i++) {
        scanf("%d%d%d", &u, &v, &val);
        vec[u].push_back(p(val, v));
        vec[v].push_back(p(val, u));
    }
    priority_queue<p, vector<p>, greater<p>> pq;
    vis[1] = 1;
    for (int i = 0; i < vec[1].size();i++)
        pq.push(vec[1][i]);
    int ans = 0;
    while (!pq.empty()) {
        p now = pq.top();
        pq.pop();
        if (!vis[now.second]) {
            ans += now.first;
            vis[now.second] = 1;
        }
        for (int i = 0; i < vec[now.second].size();i++) {
            if (!vis[vec[now.second][i].second])
                pq.push(vec[now.second][i]);
        }
    }
    printf("%d\n", ans);
    return 0;
}   
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值