POJ3255 Roadblocks ——次短路

14 篇文章 0 订阅
该博客主要介绍了如何利用Dijkstra算法同时计算图中两个节点之间的最短路径和次短路径。代码示例展示了如何在C++中实现这一过程,包括边的存储、最短路径和次短路径数组的初始化,以及优先队列的使用来不断更新路径长度。通过输入图的节点数、边数和边的权重,程序能够找出从源节点到所有其他节点的最短路径和次短路径,并输出次短路径的长度。
摘要由CSDN通过智能技术生成

Roadblocks

题意:求次短路

思路:见注释,用dijkstra同时维护最短路和次短路

// Decline is inevitable
// Romance will last forever
// UVA 10917
#include <bits/stdc++.h>
#define mst(a, x) memset(a, x, sizeof(a))
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 5e3 + 10;
const int maxm = 2e5 + 10;
struct Edge {
    int to, dis, next;
}edge[maxm];
int n, m;
int head[maxn], dis[maxn], cnt; //最短路
int dis2[maxn]; //次短路

struct node {
    int dis;
    int pos;
    node(int _dis = 0, int _pos = 0) : dis(_dis), pos(_pos) {}
    bool operator <(const node &x) const {
        return dis > x.dis;
    }
};
void add_edge(int u, int v, int w) {
    cnt++;
    edge[cnt].to = v;
    edge[cnt].dis = w;
    edge[cnt].next = head[u];
    head[u] = cnt;
}
priority_queue<node> q;
void dijkstra(int s) {
    mst(dis, 0x3f);
    mst(dis2, 0x3f);
    while(!q.empty()) q.pop();
    dis[s] = 0;
    q.push(node(0, s));
    while(!q.empty()) {
        node tmp = q.top();
        q.pop();
        int x = tmp.pos;
        int p = tmp.dis;
        if(dis2[x] < p) continue;   //如果p比第二短路都大,那么就可以直接跳过
        for(int i = head[x]; i; i = edge[i].next) {
            int y = edge[i].to;
            int d2 = p + edge[i].dis; //注意松弛操作不能d2 = dis[x] + edge[i].dis;
            //只能用tmp.dis, 要不然就无法用次短路更新次短路
            if(d2 < dis[y])     //如果这条路比最短路短 更新最短路
                dis[y] = d2,
                q.push(node(dis[y], y));
            if(dis[y] < d2 && d2<dis2[y])
                dis2[y] = d2,       //若比最短路长,次短路短,则更新次短路
                q.push(node(dis2[y], y));   //记得把这个次短路也加入队列
            
        }
    }
}
int main() {
    cin >> n >> m;
    for(int i = 1; i <= m; i++) {
        int u, v, w;
        cin >> u >> v >> w;
        add_edge(u, v, w),
        add_edge(v, u, w);
    }
    dijkstra(1);
    cout << dis2[n] << endl;
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值