最短路和次短路問題,dijkstra算法

/* 
 *題目大意： 
 *在一個有向圖中,求從s到t兩個點之間的最短路和比最短路長1的次短路的條數之和; 
 * 
 *算法思想： 
 *用A*求第K短路,目測會超時,直接在dijkstra算法上求次短路; 
 *將dist數組開成二維的,即dist[v][2],第二維分別用於記錄最短路和次短路; 
 *再用一個cnt二維數組分別記錄最短路和次短路的條數; 
 *每次更新路徑的條數時,不能直接加1,,應該加上cnt[u][k],k為次短路徑或者最短路徑的標記; 
 *圖有重邊,不能用鄰接矩陣存儲; 
 *不知道為什么,題目上說的是N and M, separated by a single space, with 2≤N≤1000 and 1 ≤ M ≤ 10000; 
 *而我的代碼硬是把N開成1W了才過,求解釋,RE了無數次,擦; 
**/ 
#include<iostream>  
#include<cstdio>  
#include<cstring>  
#include<string>  
#include<algorithm>  
using namespace std;  
   
const int N=11111;  
const int M=111111;  
const int INF=0xffffff;  
   
struct node  
{  
    int to;  
    int w;  
    int next;  
};  
   
node edge[N];  
int head[N];  
   
int dist[N][2],cnt[N][2];  
bool vis[N][2];  
int n,m,s,t,edges;  
   
void addedge(int u,int v,int w)  
{  
    edge[edges].w=w;  
    edge[edges].to=v;  
    edge[edges].next=head[u];  
    head[u]=edges++;  
}  
   
int dijkstra()  
{  
    int k;  
    for(int i=0; i<=n; i++)  
    {  
        dist[i][0]=dist[i][1]=INF;  
        vis[i][0]=vis[i][1]=0;  
        cnt[i][0]=cnt[i][1]=0;  
    }  
    cnt[s][0]=1,dist[s][0]=0;  
   
    for(int i=1; i<=n*2; i++)  
    {  
        int u=-1;  
        int min_dist=INF;  
        for(int j=1; j<=n; j++)  
            for(int flag=0; flag<2; flag++)  
                if(!vis[j][flag]&&dist[j][flag]<min_dist)  
                {  
                    min_dist=dist[j][flag];  
                    u=j;  
                    k=flag;  
                }  
        if(u==-1)  
            break;  
        vis[u][k]=true;  
        for(int e=head[u]; e!=-1; e=edge[e].next)  
        {  
            int j=edge[e].to;  
            int tmp=dist[u][k]+edge[e].w;  
   
            if(tmp<dist[j][0])//tmp小於最短路徑長:  
            {  
                dist[j][1]=dist[j][0];//次短路徑長  
                cnt[j][1]=cnt[j][0];//次短路徑計數  
                dist[j][0]=tmp;//最短路徑長  
                cnt[j][0]=cnt[u][k];//最短路徑計數  
            }  
   
            else if(tmp==dist[j][0])//tmp等於最短路徑長：  
            {  
                cnt[j][0]+=cnt[u][k];//最短路徑計數  
            }  
   
            else if(tmp<dist[j][1])//tmp大於最短路徑長且小於次短路徑長：  
            {  
                dist[j][1]=tmp;//次短路徑長  
                cnt[j][1]=cnt[u][k];//次短路徑計數  
            }  
   
            else if(tmp==dist[j][1])//tmp等於次短路徑長：  
            {  
                cnt[j][1]+=cnt[u][k];//次短路徑計數  
            }  
        }  
    }  
   
    int res=cnt[t][0];  
    if(dist[t][0]+1==dist[t][1])//判斷最短路和次短路是否相差1  
        res+=cnt[t][1];  
    return res;  
}  
   
int main()  
{  
    //freopen("C:\\Users\\Administrator\\Desktop\\kd.txt","r",stdin);  
    int tcase;  
    scanf("%d",&tcase);  
    while(tcase--)  
    {  
        edges=0;  
        scanf("%d%d",&n,&m);  
        memset(head,-1,sizeof(head));  
        int u,v,w;  
        for(int i=0; i<m; i++)  
        {  
            scanf("%d%d%d",&u,&v,&w);  
            addedge(u,v,w);  
        }  
        scanf("%d%d",&s,&t);  
        printf("%d\n",dijkstra());  
    }  
    return 0;  
}