最短路及次短路

#include<iostream>
#include<cstdio>
#include<cmath>
#include<map>
#include<queue>
#include<vector>
#include<cstring>
#include<algorithm>
#define rep(i,a,b) for(int i=(a);i<(b);i++)
#define rev(i,a,b) for(int i=(a);i>=(b);i--)
#define clr(a,x) memset(a,x,sizeof a)
#define inf 0x3f3f3f3f
using namespace std;

const int maxn=1010;
const int maxm=10005;
struct node
{
    int u,v,w;
    node(int uu,int vv,int ww)
    {
        u=uu;v=vv;w=ww;
    }
};
int vis[2][maxn],cnt[2][maxn],dis[2][maxn];
vector<node>g[maxn];
int n,m;
int dijkstra(int s,int t)
{
    clr(vis,0);
    clr(dis,0x3f);
    clr(cnt,0);
    dis[0][s]=0;
    cnt[0][s]=1;
    int flag=0;
    for(int i=1;i<=2*n;i++)
    {
        int minx=inf,k;
        for(int j=1;j<=n;j++)
        {
            if(!vis[0][j]&&dis[0][j]<minx)
            {
                flag=0;
                minx=dis[0][j];k=j;
            }
            else if(!vis[1][j]&&dis[1][j]<minx)
            {
                flag=1;
                minx=dis[1][j];k=j;
            }
        }
        if(minx==inf)break;
        vis[flag][k]=1;
        for(int j=0;j<g[k].size();j++)
        {
            int v=g[k][j].v;
            int w=g[k][j].w;
            if(minx+w<dis[0][v])
            {
                dis[1][v]=dis[0][v];
                cnt[1][v]=cnt[0][v];
                dis[0][v]=minx+w;
                cnt[0][v]=cnt[flag][k];
            }
            else if(minx+w==dis[0][v])
            {
                cnt[0][v]+=cnt[flag][k];
            }
            else if(minx+w<dis[1][v])
            {
                dis[1][v]=minx+w;
                cnt[1][v]=cnt[flag][k];
            }
            else if(minx+w==dis[1][v])
            {
                cnt[1][v]+=cnt[flag][k];
            }
        }
    }
    int sum=cnt[0][t];
    if(dis[1][t]-1==dis[0][t])sum+=cnt[1][t];
    return sum;
}
int main()
{
    int t,a,b,c;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)g[i].clear();
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            g[a].push_back(node(a,b,c));
        }
        scanf("%d%d",&a,&b);
        printf("%d\n",dijkstra(a,b));
    }
    return 0;
}