POJ 3463 Dijkstra 最短路 次短路 计数统计

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<queue>
using namespace std;
const int maxe=20010;
const int maxv=1010;
const int INF=0x3f3f3f3f;
int head[maxv],cnt,dis[maxv][2],ccnt[maxv][2];
bool visit[maxv][2];
struct Edge {
    int to,val,next;
} edge[maxe];
struct Node {
    int u,dis;
    bool operator< (const Node rhs) const {
        return this->dis>rhs.dis;
    }
} now,temp;
void init() {
    memset(head,-1,sizeof head);
    cnt=0;
}
void addedge(int from,int to,int val) {
    edge[cnt].to=to;
    edge[cnt].val=val;
    edge[cnt].next=head[from];
    head[from]=cnt++;
}
void Dijkstra(int src,int des,int n) {
    for(int i=1; i<=n; i++) {
        dis[i][0]=INF,dis[i][1]=INF;
        visit[i][0]=false,visit[i][1]=false;
        ccnt[i][0]=0,ccnt[i][1]=0;
    }
    int k;
    bool flag;
    dis[src][0]=0,ccnt[src][0]=1;
    for(int i=1; i<=2*n-1; i++) { //找到当前位置
        int mmin=INF;
        for(int j=1; j<=n; j++) {
            if(!visit[j][0]&&mmin>dis[j][0])
                k=j,flag=0,mmin=dis[j][0];
            else if(!visit[j][1]&&mmin>dis[j][1])
                k=j,flag=1,mmin=dis[j][1];
        }
        if(mmin==INF)
            break;
        visit[k][flag]=1;
        for(int j=head[k]; j!=-1; j=edge[j].next) {
            int v=edge[j].to;
            if(dis[v][0]>mmin+edge[j].val) {
                dis[v][1]=dis[v][0];//当前的最短路等于现在的次短路，明白这个逻辑就可以明白为什么Dijkstra可以求次短路了
                ccnt[v][1]=ccnt[v][0];//计数转移
                dis[v][0]=mmin+edge[j].val;//最短路更新
                ccnt[v][0]=ccnt[k][flag];//计数更新
            } else if(dis[v][0]==mmin+edge[j].val) {
                ccnt[v][0]+=ccnt[k][flag];
            } else if(dis[v][1]>mmin+edge[j].val) {
                dis[v][1]=mmin+edge[j].val;
                ccnt[v][1]=ccnt[k][flag];
            } else if(dis[v][1]==mmin+edge[j].val) {
                ccnt[v][1]+=ccnt[k][flag];
            }
        }
    }
}
int main(void) {
#ifndef ONLINE_JUDGE
    freopen("E:\\input.txt","r",stdin);
#endif // ONLINE_JUDGE
    int n,m,T,t1,t2,t3,src,des;
    scanf("%d",&T);
    while(T--) {
        scanf("%d%d",&n,&m);
        init();
        while(m--) {
            scanf("%d%d%d",&t1,&t2,&t3);
            addedge(t1,t2,t3);
        }
        scanf("%d%d",&src,&des);
        Dijkstra(src,des,n);
        if(dis[des][1]==dis[des][0]+1)//次短路的距离恰好比最短路少一
            printf("%d\n",ccnt[des][0]+ccnt[des][1]);
        else//次短路不满足条件 直接输出最短路
            printf("%d\n",ccnt[des][0]);
    }
    return 0;
}