Sightseeing - statistics of the number of shortest paths + state transfer + secondary small paths

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
#define x first
#define y second
typedef pair<int, int> PII;
const int N = 1010,M = 10010;
vector<PII>g[N];
int n,m,s,t;
int dist[N][2];
int cnt[N][2];
bool st[N][2];
struct node
{
    
    int ver,type,w;
    bool operator > (const node &W)const
    {
    
        return w>W.w;
    }
};
int dij()
{
    
    memset(dist,0x3f,sizeof dist);
    memset(cnt,0,sizeof cnt);
    memset(st,0,sizeof st);
    dist[s][0]=0;cnt[s][0]=1;
    priority_queue<node,vector<node>,greater<node> >heap;
    heap.push({
    s,0,0});
    while(heap.size())
    {
    
        auto t=heap.top();
        heap.pop();
        int u=t.ver,type=t.type,distance=t.w;
        if(st[u][type])continue;
        st[u][type]=true;
        for(auto [v,w] : g[u])
        {
    
            if(dist[v][0]>distance+w)
            {
    
                dist[v][1]=dist[v][0],cnt[v][1]=cnt[v][0];
                heap.push({
    v,1,dist[v][1]});
                dist[v][0]=distance+w,cnt[v][0]=cnt[u][type];
                heap.push({
    v,0,dist[v][0]});
            }
            else if(dist[v][0]==distance+w)
            {
    
                cnt[v][0]+=cnt[u][type];
            }
            else if(dist[v][1]>distance+w)
            {
    
                dist[v][1]=distance+w,cnt[v][1]=cnt[u][type];
                heap.push({
    v,1,dist[v][1]});
            }
            else if(dist[v][1]==distance+w) 
            {
    
                cnt[v][1]+=cnt[u][type];
            }
        }
    }
    int res=cnt[t][0];
    if(dist[t][0]==dist[t][1]-1)res+=cnt[t][1];
    return res;
}
signed main()
{
    
    int T;
    cin>>T;
    while(T--)
    {
    
        cin>>n>>m;
        int a,b,c;
        for(int i=1;i<=n;i++)g[i].clear();
        for(int i=1;i<=m;i++)
        {
    
            cin>>a>>b>>c;
            g[a].push_back({
    b,c});
        }
        cin>>s>>t;
        cout<<dij()<<endl;
    }
}