BZOJ 1726 严格次短路

不说什么了，模板题

题目数据有点弱，不严格的次短路也能ac。。。。

#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <iostream>
#include <cstdio>

#define N 100000
#define M 2000000

using namespace std;

struct HP
{
    int x,d;
}hp[M];

int head[N],next[M],dis[N],to[M],len[M],cnt,n,qk,m,q[M],cs[N],size;
bool vis[N];

inline void add(int u,int v,int w)
{
    to[cnt]=v; len[cnt]=w; next[cnt]=head[u]; head[u]=cnt++;
}

inline bool cmp(const HP &a,const HP &b)
{
    return a.d+dis[a.x]>b.d+dis[b.x];
}

void read()
{
    memset(head,-1,sizeof head); cnt=0;
    for(int i=1,a,b,c;i<=m;i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c); add(b,a,c);
    }
    qk=2;
}

void spfa()
{
    memset(dis,0x3f,sizeof dis);
    int h=1,t=2,sta;
    q[1]=n; vis[n]=true; dis[n]=0;
    while(h<t)
    {
        sta=q[h++];
        vis[sta]=false;
        for(int i=head[sta];~i;i=next[i])
            if(dis[to[i]]>dis[sta]+len[i])
            {
                dis[to[i]]=dis[sta]+len[i];
                if(!vis[to[i]]) vis[to[i]]=true,q[t++]=to[i];
            }
    }
}

int bfs()
{
    memset(cs,0,sizeof cs);
    bool fg=false;int mindis=-1;
    hp[1].x=1; hp[1].d=0; size=1;
    while(size)
    {
        HP sta=hp[1];
        pop_heap(hp+1,hp+1+size,cmp); size--;
        
        if(sta.x!=n) cs[sta.x]++;
        else if(sta.d!=mindis) cs[sta.x]++;
        
        if(sta.x==n&&!fg)
        {
            mindis=sta.d;
            fg=true;
        }
        if(cs[n]==qk) return sta.d;
        if(cs[sta.x]>qk) continue;
        for(int i=head[sta.x];~i;i=next[i])
        {
            size++;
            hp[size].x=to[i]; hp[size].d=sta.d+len[i];
            push_heap(hp+1,hp+1+size,cmp);
        }
    }
}

void go()
{
    spfa();
    if(1==n) qk++;
    printf("%d\n",bfs());
}

int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        read();
        go();
    }
    return 0;
}