POJ 3463 有向图求次短路的长度及其方法数

题目大意：

希望求出走出最短路的方法总数，如果次短路只比最短路小1，那也是可取的

输出总的方法数

这里n个点，每个点有最短和次短两种长度

这里采取的是dijkstra的思想，相当于我们可以不断找到更新到的最短长度来更新其他长度，保证之前的所有可取的最短长度都已经更新的情况下，这样是除了第一个点的最短路为0已知，还需要更新2*n-1次，如果从一个点的位置出发更新了其他点，那么这个位置就不再作为可更新点～～这里都是暴力找最优的可更新的点～～不知道如何做到像普通的dijkstra那种log级别的找点～～

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 using namespace std;
 #define N 1005
 #define M 10005
 #define clr(a,b) memset(a,b,sizeof(a))
 struct Edge{
     int x,y,d,next;
     Edge(int x=,int y=,int d=,int next=):x(x),y(y),d(d),next(next){}
 }e[M<<];
 int first[N] , tot , n , m , s , t;
 void add_edge(int x,int y,int d)
 {
     e[tot] = Edge(x,y,d,first[x]);
     first[x] = tot++;
 }
 int dis[N][] , cnt[N][] , vis[N][];
 void dijkstra(int s , int t)
 {
     clr(dis,0x3f);clr(vis,);clr(cnt,);
     dis[s][]=;cnt[s][]=;
     for(int i=;i<*n;i++){
         int a= , b= , curd=0x7fffffff;
         for(int j=;j<=n;j++){
             if(dis[j][]<curd && !vis[j][]) curd=dis[j][] , a=j,b=;
             if(dis[j][]<curd && !vis[j][]) curd=dis[j][] , a=j,b=;
         }
        // cout<<a<<" "<<b<<" "<<curd<<endl;
         vis[a][b]=;
         for(int i=first[a];~i;i=e[i].next){
             int v = e[i].y , nowd=curd+e[i].d;
             if(nowd<dis[v][]){
                 dis[v][]=dis[v][];cnt[v][]=cnt[v][];
                 dis[v][]=nowd;cnt[v][]=cnt[a][b];
             }
             else if(nowd == dis[v][]){
                 cnt[v][]+=cnt[a][b];
             }
             else if(nowd <dis[v][]){
                 dis[v][]=nowd ;cnt[v][]=cnt[a][b];
             }
             else if(nowd==dis[v][]){
                 cnt[v][]+=cnt[a][b];
             }
         }
     }
   //  cout<<cnt[t][0]<<" "<<cnt[t][1]<<endl;
     if(dis[t][]- == dis[t][]) cnt[t][]+=cnt[t][];
     printf("%d\n" , cnt[t][]);
 }
 int main()
 {
    // freopen("a.in" , "r" , stdin);
     int T;
     scanf("%d" , &T);
     while(T--){
         scanf("%d%d" , &n , &m);
         clr(first,-);tot=;
         for(int i=; i<m ; i++){
             int x,y,d;
             scanf("%d%d%d" , &x,  &y , &d);
             add_edge(x , y , d);
         }

         scanf("%d%d" , &s , &t);
         dijkstra(s,t);
     }

     return ;
 }