P2865 [USACO06NOV]路障Roadblocks

P2865 [USACO06NOV]路障Roadblocks

最短路（次短路）

直接在dijkstra中维护2个数组：d1（最短路）,d2（次短路），然后跑一遍就行了。

attention：数据有不同权值的重边（40ptsQAQ）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cctype>
using namespace std;
template <typename T> inline void read(T &x){
    char c=getchar(); x=0;
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) x=(x<<3)+(x<<1)+(c^48),c=getchar();
}
struct data{
    int d,u;
    bool operator < (const data &tmp) const {return d>tmp.d;}
}; priority_queue <data> h;
int n,m,cnt,d1[5002],d2[5002],hd[5002],nxt[200002],ed[5002],poi[200002],val[200002];
inline void add(int x,int y,int v){
    nxt[ed[x]]=++cnt; hd[x]=hd[x] ? hd[x]:cnt;
    ed[x]=cnt; poi[cnt]=y; val[cnt]=v;
}
int main(){
    read(n); read(m); int q1,q2,q3;
    for(int i=1;i<=m;++i) read(q1),read(q2),read(q3),add(q1,q2,q3),add(q2,q1,q3);
    memset(d1,127,sizeof(d1));
    memset(d2,127,sizeof(d2));
    h.push((data){d1[1]=0,1});
    while(!h.empty()){
        data x=h.top(); h.pop();
        if(x.d!=d1[x.u]&&x.d!=d2[x.u]) continue;
        for(int i=hd[x.u];i;i=nxt[i]){
            if(x.d+val[i]<d1[poi[i]]){ //最短路更新
                d2[poi[i]]=d1[poi[i]]; 
                d1[poi[i]]=x.d+val[i];
                h.push((data){d1[poi[i]],poi[i]});
            }else if(x.d+val[i]<d2[poi[i]]){ //次短路更新
                d2[poi[i]]=x.d+val[i];
                h.push((data){d2[poi[i]],poi[i]});
            }
        }
    }printf("%d",d2[n]);
    return 0;
}