poj3225 Roadblocks

题意:某道路共有R条道路,N个路口。问从1号路口到N路口的次短路是?次短路是比最短路长度次短的路径。
思路:Dijkstra 算法虽然只能求最短路,但是我们修改一下。次短路:要么是到终点的前某一个点的最短路加上它到终点的路,或者是要么到它的次短路加上到终点的最短路。因此,对于每个顶点,我们记录的不仅是最短路,还有 次短路距离。接下来就是和Dijkstra()一样的做法,不断更新这两个距离就可以求出次短路。
Dijkstra用的是优先队列优化过的,挑战代码真精炼,可以直接当模板用了。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
const int N=5050;
typedef pair<int,int> P;
struct edge{
 int to,w;
};
vector<edge>G[N];
int dist[N],dist2[N],n,m;
const int inf=2<<29;
void solve()
{
    priority_queue<P,vector<P>,greater<P> >que;
    for(int i=1;i<=n;i++) dist[i]=inf;
    for(int i=1;i<=n;i++) dist2[i]=inf;
    dist[1]=0;
    que.push(P(0,1));
    while(!que.empty()){
        P now=que.top();
        que.pop();
        int v=now.second,d=now.first;
        if(dist2[v]<d) continue;
        for(int i=0;i<G[v].size();i++){
            edge &e=G[v][i];
            int d2=d+e.w;
            if(dist[e.to]>d2){
                swap(dist[e.to],d2);
                que.push(P(dist[e.to],e.to));
            }
            if(dist2[e.to]>d2){
                swap(dist2[e.to],d2);
                que.push(P(dist2[e.to],e.to));
            }
        }
    }
    printf("%d\n",dist2[n]);
}

int main()
{
   while(~scanf("%d%d",&n,&m)){
      for(int i=0;i<m;i++){
        int a,b,c;scanf("%d%d%d",&a,&b,&c);
        G[a].push_back(edge{b,c});
        G[b].push_back(edge{a,c});
      }
      solve();
   }
}

你可能感兴趣的:(poj3225 Roadblocks)