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=0,int y=0,int d=0,int next=0):x(x),y(y),d(d),next(next){}
}e[M<<1];
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][2] , cnt[N][2] , vis[N][2];
void dijkstra(int s , int t)
{
    clr(dis,0x3f);clr(vis,0);clr(cnt,0);
    dis[s][0]=0;cnt[s][0]=1;
    for(int i=1;i<2*n;i++){
        int a=0 , b=0 , curd=0x7fffffff;
        for(int j=1;j<=n;j++){
            if(dis[j][0]<curd && !vis[j][0]) curd=dis[j][0] , a=j,b=0;
            if(dis[j][1]<curd && !vis[j][1]) curd=dis[j][1] , a=j,b=1;
        }
       // cout<<a<<" "<<b<<" "<<curd<<endl;
        vis[a][b]=1;
        for(int i=first[a];~i;i=e[i].next){
            int v = e[i].y , nowd=curd+e[i].d;
            if(nowd<dis[v][0]){
                dis[v][1]=dis[v][0];cnt[v][1]=cnt[v][0];
                dis[v][0]=nowd;cnt[v][0]=cnt[a][b];
            }
            else if(nowd == dis[v][0]){
                cnt[v][0]+=cnt[a][b];
            }
            else if(nowd <dis[v][1]){
                dis[v][1]=nowd ;cnt[v][1]=cnt[a][b];
            }
            else if(nowd==dis[v][1]){
                cnt[v][1]+=cnt[a][b];
            }
        }
    }
  //  cout<<cnt[t][0]<<" "<<cnt[t][1]<<endl;
    if(dis[t][1]-1 == dis[t][0]) cnt[t][0]+=cnt[t][1];
    printf("%d\n" , cnt[t][0]);
}
int main()
{
   // freopen("a.in" , "r" , stdin);
    int T;
    scanf("%d" , &T);
    while(T--){
        scanf("%d%d" , &n , &m);
        clr(first,-1);tot=0;
        for(int i=0; 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 0;
}