POJ 3255 Roadblocks（次短路）

Roadblocks

题目链接：

http://poj.org/problem?id=3255

解题思路：

Dijkstra算法的思路是依次确定尚未确定的顶点中距离最小的顶点。按照这个思路对算法进行少许修改，就可以简单得求出次短路

了。到某个顶点v的次短路要么是到其他某个顶点u的最短路再加上u->v的边，要么是到u的次短路再加上u->v的边，因此所需要求的

就是到所有顶点的最短路和次短路。因此，对于每个顶点，我们记录的不仅仅是最短距离，还有次短的距离。接下来只要用与

Dijkstra算法相同的做法，不断更新这两个距离就可以求出次短路了。

AC代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>
#define INF 0xfffffff
using namespace std;

struct Edge{
    int to,cost;
};
int n,r;
vector<Edge> edge[5005];
typedef pair<int,int> P;//first是最短距离，second是顶点编号
int dis1[5005];//最短距离
int dis2[5005];//次短距离

void solve(){
    priority_queue<P,vector<P>,greater<P> > q;
    for(int i = 0; i <= n; i++){
        dis1[i] = INF;
        dis2[i] = INF;
    }
    dis1[1] = 0;
    q.push(P(0,1));
    while(!q.empty()){
        P p = q.top();
        q.pop();
        int v = p.second,d = p.first;
        if(dis2[v] < d)
            continue;
        for(int i = 0; i < edge[v].size(); i++){
            Edge &e = edge[v][i];
            int d2 = d + e.cost;
            if(dis1[e.to] > d2){
                swap(dis1[e.to],d2);
                q.push(P(dis1[e.to],e.to));
            }
            if(dis2[e.to] > d2 && dis1[e.to] < d2){
                dis2[e.to] = d2;
                q.push(P(dis2[e.to],e.to));
            }
        }
    }
    printf("%d\n",dis2[n]);
}

int main(){
    while(~scanf("%d%d",&n,&r)){
        for(int i = 0; i <= n; i++)
            edge[i].clear();
        int u,v,cost;
        Edge tmp;
        for(int i = 0; i < r; i++){
            scanf("%d%d%d",&u,&v,&cost);
            tmp.to = v;tmp.cost = cost;
            edge[u].push_back(tmp);
            tmp.to = u;
            edge[v].push_back(tmp);
        }
        solve();
    }
    return 0;
}