POJ3255---求次短路（用Dijkstra）

来源：互联网发布：seo 关键字编辑：程序博客网时间：2024/05/26 15:59

题目链接

分析：百度翻译一下题目发现是道求最短路且可以重复经过某点的情况，可以用Dijkstra求，具体做法：用 dist1[] 记录最短路，dist2[] 记录次短路。然后还是 spfa, 更新的时候分情况判断一下，次短路可能更新次短路，最短路可能更新次短路和最短路。。。记的更新完后把dist2[]也要加进去。

#include<iostream>#include<cstring>#include<set>using namespace std;const int maxn=5010;const int maxm=400010;const int inf=0x3f3f;int n,m;int p[maxn],eid,pre[maxn];struct edge{    int v,next,w;}e[maxm];void init(){    memset(p,-1,sizeof(p));    eid=0;}void insert(int u,int v,int w){    e[eid].v=v;    e[eid].next=p[u];    e[eid].w=w;    p[u]=eid++;}void insert2(int u,int v,int w){    insert(u,v,w);    insert(v,u,w);}typedef pair<int,int>PII;set<PII>min_heap;int dist[maxn],dist2[maxn];bool vst[maxn];bool dijkstra(int v){    memset(vst,0,sizeof(vst));    memset(dist,0x3f,sizeof(dist));    memset(dist2,0x3f,sizeof(dist2));    dist[v]=0;    min_heap.insert(make_pair(0,v));    for(int i=0;i<2*n;i++)    {        if(min_heap.size()==0)return false;        set<PII>::iterator iter=min_heap.begin();        int  j=iter->second;        min_heap.erase(*iter);        vst[j]=true;        for(int x=p[j];x!=-1;x=e[x].next){            int f=e[x].v;            if(dist[j]+e[x].w>dist[f]&&dist[j]+e[x].w<dist2[f]){                min_heap.erase(make_pair(dist2[f],f));                dist2[f]=dist[j]+e[x].w;                min_heap.insert(make_pair(dist2[f],f));            }            if(dist2[j]+e[x].w<dist2[f]){                min_heap.erase(make_pair(dist2[f],f));                dist2[f]=dist2[j]+e[x].w;                min_heap.insert(make_pair(dist2[f],f));            }            if(!vst[f]&&dist[j]+e[x].w<dist[f]){                min_heap.erase(make_pair(dist[f],f));                dist[f]=dist[j]+e[x].w;                min_heap.insert(make_pair(dist[f],f));            }        }    }    return true;}int main(){    init();    cin>>n>>m;    for(int i=0;i<m;i++){        int a,b,c;        cin>>a>>b>>c;        insert2(a,b,c);    }    dijkstra(1);    cout<<dist2[n]<<endl;}

阅读全文

0 0