acwing383. 观光

题目

“您的个人假期”旅行社组织了一次比荷卢经济联盟的巴士之旅。

比荷卢经济联盟有很多公交线路。

每天公共汽车都会从一座城市开往另一座城市。

沿途汽车可能会在一些城市（零或更多）停靠。

旅行社计划旅途从 S 城市出发，到 F 城市结束。

由于不同旅客的景点偏好不同，所以为了迎合更多旅客，旅行社将为客户提供多种不同线路。

游客可以选择的行进路线有所限制，要么满足所选路线总路程为 S 到 F 的最小路程，要么满足所选路线总路程仅比最小路程多一个单位长度。

如上图所示，如果 S=1，F=5，则这里有两条最短路线 1→2→5,1→3→5，长度为 6；有一条比最短路程多一个单位长度的路线 1→3→4→5，长度为 7。

现在给定比荷卢经济联盟的公交路线图以及两个城市 S 和 F，请你求出旅行社最多可以为旅客提供多少种不同的满足限制条件的线路。

输入格式

第一行包含整数 T，表示共有 T 组测试数据。

每组数据第一行包含两个整数 N 和 M，分别表示总城市数量和道路数量。

接下来 M 行，每行包含三个整数 A,B,L，表示有一条线路从城市 A 通往城市 B，长度为 L。

需注意，线路是 单向的，存在从 A 到 B 的线路不代表一定存在从 B 到 A 的线路，另外从城市 A 到城市 B 可能存在多个不同的线路。

接下来一行，包含两个整数 S 和 F，数据保证 S 和 F 不同，并且 S、F 之间至少存在一条线路。

输出格式

每组数据输出一个结果，每个结果占一行。

数据保证结果不超过 109。

数据范围

2≤N≤1000,
1≤M≤10000,
1≤L≤1000，
1≤A,B,S,F≤N

输入样例：

2
5 8
1 2 3
1 3 2
1 4 5
2 3 1
2 5 3
3 4 2
3 5 4
4 5 3
1 5
5 6
2 3 1
3 2 1
3 1 10
4 5 2
5 2 7
5 2 7
4 1

输出样例：

3
2

题解

我们可以存下当前的最短路和次短路的值，在迪杰斯特拉中更新这两个值：

• 如果当前值比最短路值还大，那么次短路更新为最短路的值，并将次短路的条数等于最短路的条数 cnt[j][1]=cnt[j][0];并更新当前节点的最短路，最后进队
• 如果当前值等于最短路的值，说明找到另外一条通路，将两次最短路的条数相加，cnt[j][0]+=count;之后进队
• 如果当前值比次短路的值大，更新当前节点的次短路，并进队
• 如果当前值等于次短路的值，则次短路的条数累加 cnt[j][1]+=count;

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=1010,M=20010;
int dist[N][2],cnt[N][2];
int h[N],w[M],ne[M],e[M],idx;
struct Ver{
    int id,type,dist;
    bool operator>(const Ver &W)const{
        return dist>W.dist;
    }
};
int T,n,m,s,f;
bool st[N][2];
void add(int a,int b,int c){
    e[idx]=b;
    ne[idx]=h[a];
    w[idx]=c;
    h[a]=idx++;
}
int dijkstra(){
    memset(st,0,sizeof st);
    memset(dist,0x3f,sizeof dist);
    memset(cnt,0,sizeof cnt);
    dist[s][0]=0;
    cnt[s][0]=1;
    priority_queue<Ver,vector<Ver>,greater<Ver>>heap;
    heap.push({s,0,0});
    while(heap.size()){
        auto t=heap.top();
        heap.pop();
        int ver=t.id,distence=t.dist,type=t.type,count=cnt[ver][type];
        if(st[ver][type])continue;
        st[ver][type]=true;
        for(int i=h[ver];~i;i=ne[i]){
            int j=e[i];
            if(dist[j][0]>distence+w[i]){
                dist[j][1]=dist[j][0],cnt[j][1]=cnt[j][0];
                heap.push({j,1,dist[j][1]});
                dist[j][0]=distence+w[i],cnt[j][0]=count;
                heap.push({j,0,dist[j][0]});
            }else if(dist[j][0]==distence+w[i]){
                cnt[j][0]+=count;
            }else if(dist[j][1]>distence+w[i]){
                dist[j][1]=distence+w[i];
                cnt[j][1]=count;
                heap.push({j,1,dist[j][1]});
            }else if(dist[j][1]==distence+w[i])cnt[j][1]+=count;
        }
    }
    int ans=cnt[f][0];
    if(dist[f][0]+1==dist[f][1])ans+=cnt[f][1];
    return ans;
}

int main(){
    scanf("%d",&T);
    while(T--){
        idx=0;
        memset(h,-1,sizeof h);
        scanf("%d%d",&n,&m);
        while(m--){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
        }
        scanf("%d%d",&s,&f);
        printf("%d\n",dijkstra());
    }
    
    
    return 0;
}