观光（最短路和次短路计数）

传送门：观光

思路：这里因为是有边权的最短路，使用堆优化的dijkstra，保证每个顶点的次短路都只会被入队出队一次，又因为每次取出来的都是dist最小的一个，具有拓扑性质。

代码：

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
typedef pair<int,int>PII;
const int N=2e4+10;

int n,m,s,t;
bool st[N][2];
int dist[N][2],cnt[N][2];
int idx,e[N*2],ne[N*2],h[N],w[2*N];
struct Ver
{
    int id,type,dist;
   bool operator> (const Ver &W) const
    {
        return dist > W.dist;
    }
};
void  add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int dijkstra()
{
   memset(dist,0x3f,sizeof dist);
   memset(st,0,sizeof st);
   memset(cnt,0,sizeof cnt);

   dist[s][0]=0;
   cnt[s][0]=1;

   priority_queue<Ver, vector<Ver>, greater<Ver>> heap;
   heap.push({s,0,0});

   while(heap.size())
   {
       Ver p=heap.top();
       heap.pop();
       int ver=p.id,type=p.type,distance=p.dist,cont=cnt[ver][type];
       if(st[ver][type]) continue;
       st[ver][type]=true;
       for(int i=h[ver];i!=-1;i=ne[i])
       {
           int j=e[i];
           if(dist[j][0]>distance+w[i])
           {
               dist[j][1]=dist[j][0],cnt[j][1]=cnt[j][0];
               heap.push({j,1,dist[j][1]});
               dist[j][0]=distance+w[i],cnt[j][0]=cont;
               heap.push({j,0,dist[j][0]});

           }else if(dist[j][0]==distance+w[i]) cnt[j][0]+=cont;
           else if(dist[j][1]>distance+w[i])
           {
               dist[j][1]=distance+w[i], cnt[j][1]=cont;
                heap.push({j, 1, dist[j][1]});
           }else if(dist[j][1]==distance+w[i]) cnt[j][1]+=cont;
       }
   }
   int res=cnt[t][0];
   if(dist[t][0]+1==dist[t][1]) res+=cnt[t][1];
   return res;
}
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        cin>>n>>m;
        memset(h,-1,sizeof h);
        idx=0;
        while(m--)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
           add(a,b,c);
        }
        cin>>s>>t;
        cout<<dijkstra()<<endl;
    }
    return 0;
}