最短路和次短路问题,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=; 
 const int M=; 
 const int INF=0xffffff; 
    
 struct node 
 { 
     int to; 
     int w; 
     int next; 
 }; 
    
 node edge[N]; 
 int head[N]; 
    
 int dist[N][],cnt[N][]; 
 bool vis[N][]; 
 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=; i<=n; i++) 
     { 
         dist[i][]=dist[i][]=INF; 
         vis[i][]=vis[i][]=; 
         cnt[i][]=cnt[i][]=; 
     } 
     cnt[s][]=,dist[s][]=; 
    
     for(int i=; i<=n*; i++) 
     { 
         int u=-; 
         int min_dist=INF; 
         for(int j=; j<=n; j++) 
             for(int flag=; flag<; flag++) 
                 if(!vis[j][flag]&&dist[j][flag]<min_dist) 
                 { 
                     min_dist=dist[j][flag]; 
                     u=j; 
                     k=flag; 
                 } 
         if(u==-) 
             break; 
         vis[u][k]=true; 
         for(int e=head[u]; e!=-; e=edge[e].next) 
         { 
             int j=edge[e].to; 
             int tmp=dist[u][k]+edge[e].w; 
    
             if(tmp<dist[j][])//tmp小于最短路径长: 
             { 
                 dist[j][]=dist[j][];//次短路径长 
                 cnt[j][]=cnt[j][];//次短路径计数 
                 dist[j][]=tmp;//最短路径长 
                 cnt[j][]=cnt[u][k];//最短路径计数 
             } 
    
             else if(tmp==dist[j][])//tmp等于最短路径长： 
             { 
                 cnt[j][]+=cnt[u][k];//最短路径计数 
             } 
    
             else if(tmp<dist[j][])//tmp大于最短路径长且小于次短路径长： 
             { 
                 dist[j][]=tmp;//次短路径长 
                 cnt[j][]=cnt[u][k];//次短路径计数 
             } 
    
             else if(tmp==dist[j][])//tmp等于次短路径长： 
             { 
                 cnt[j][]+=cnt[u][k];//次短路径计数 
             } 
         } 
     } 
    
     int res=cnt[t][]; 
     if(dist[t][]+==dist[t][])//判断最短路和次短路是否相差1 
         res+=cnt[t][]; 
     return res; 
 } 
    
 int main() 
 { 
     //freopen("C:\\Users\\Administrator\\Desktop\\kd.txt","r",stdin); 
     int tcase; 
     scanf("%d",&tcase); 
     while(tcase--) 
     { 
         edges=; 
         scanf("%d%d",&n,&m); 
         memset(head,-,sizeof(head)); 
         int u,v,w; 
         for(int i=; i<m; i++) 
         { 
             scanf("%d%d%d",&u,&v,&w); 
             addedge(u,v,w); 
         } 
         scanf("%d%d",&s,&t); 
         printf("%d\n",dijkstra()); 
     } 
     return ; 
 }