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;
}