POJ-3255 Roadblocks (次短路)

题目链接：

$\quad$POJ - 3255

题目大意：

$\quad n个点，m条边，很明确地说让求 1$~$n 次短路。$

数据范围：

$\quad1 \leq n \leq 5000 \quad \quad1 \leq m \leq 100000$
$\quad 1 \leq w \leq 5000 \quad (w表示每条边的边权)$

解题思路：

$\quad次短路板子题。$
$\quad$Dijkstra算法就不再细说了。只需在Dijkstra算法内作一点修改，就可以求得次短路。
$\quad$开两个数组，$dis[i]$表示源点到$i$的最短路，$dist[i]$表示源点到$i$的次短路
$\quad$令$x$为当前距离最短的点，由它更新它所连的点$dr$，再令$D = dis[x] + cost[x, dr]。$
- $\quad 如果D \lt dis[dr]，那么在更新dr的最短路(即dis[dr])之前，先将当前的dis[dr]赋值给dist[dr]，然后再更新dis[dr];$
- $\quad如果D \gt dis[dr]$ && $D \lt dist[dr]$，那么就直接将$D赋值给dist[dr]$。
$\quad$值得注意的是，不管是更新了某个节点的最短路还是次短路后，都应该将这个节点的信息放进优先队列里，来继续更新其他的点。
$\quad$详见代码

//Dijkstra次短路
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long LL;
const int inf = 1 << 30;
const LL INF = 1LL << 60;
const int MaxN = 1e5;

int n, m, all;
int pre[2 * MaxN + 5], last[MaxN + 5], other[2 * MaxN + 5];
int cost[2 * MaxN + 5];
int dis[MaxN + 5]; //最短路
int dist[MaxN + 5]; //次短路
struct Node
{
    int id, d;
    Node () {}
    Node (int a, int b) {
        id = a; d = b;
    }
    bool friend operator < (Node x, Node y) {
        return x.d > y.d;
    }
};

void build(int x, int y, int w)
{
    pre[++all] = last[x];
    last[x] = all;
    other[all] = y;
    cost[all] = w;
}

void Dijkstra(int s)
{
    priority_queue <Node> pq;
    for(int i = 1; i <= n; i++) {
        dis[i] = inf;
        dist[i] = inf;
    }
    dis[s] = 0; 

    Node tmp = Node(1, 0);
    pq.push(tmp);

    while(!pq.empty())
    {
        Node now = pq.top();
        pq.pop();
        int ed = last[now.id], dr;
        if(dist[now.id] < now.d) continue;
        while(ed != -1) {
            dr = other[ed];
            //把dis[now.id]改为now.d就过了!
            //why!?
            //哦!知道了!dis[]存的是最短路,后面更新次短路的时候会出错!
            int D = now.d + cost[ed];

            if(D < dis[dr]) {                       //更新dis[dr]和dist[dr]
                dist[dr] = dis[dr];
                dis[dr] = D;
                tmp = Node(dr, dis[dr]);
                pq.push(tmp);
            }
            else if(D > dis[dr] && D < dist[dr]) {  //只更新dist[dr]
                dist[dr] = D;
                tmp = Node(dr, dist[dr]);
                pq.push(tmp);
            }
            ed = pre[ed];
        }
    }
}

int main()
{
    while(scanf("%d %d", &n, &m) != EOF)
    {
        all = -1;
        memset(last, -1, sizeof(last));
        for(int i = 1; i <= m; i++) {
            int u, v, w;
            scanf("%d %d %d", &u, &v, &w);
            build(u, v, w); build(v, u, w);
        }
        Dijkstra(1);
        printf("%d\n", dist[n]);
    }
    return 0;
}

这道也是次短路。