最短路计数（最短路求方案数）

题目描述

给出一个 N 个顶点 M 条边的无向无权图，顶点编号为 1 到 N。
问从顶点 1 开始，到其他每个点的最短路有几条。

输入格式

第一行包含 2 个正整数 N,M，为图的顶点数与边数。
接下来 M 行，每行两个正整数 x,y，表示有一条顶点 x 连向顶点 y 的边，请注意可能有自环与重边。

输出格式

输出 N 行，每行一个非负整数，第 i 行输出从顶点 1 到顶点 i 有多少条不同的最短路，由于答案有可能会很大，你只需要输出对 100003 取模后的结果即可。
如果无法到达顶点 i 则输出 0。

数据范围

1≤N≤105,
1≤M≤2×105

输入样例
5 7
1 2
1 3
2 4
3 4
2 3
4 5
4 5
输出样例
1
1
1
2
4

题目分析

因为要求的是方案数，因此我们在求最短路中的dist[]时，还要再额外维护一个cnt[]数组来记录最短路的方案数。

1. 当dist[v]>dist[u]+d[u][v]时，cnt[v]的值即为cnt[u]的值（因为v点最短距离是由u点的最短距离更新而来）。
2. dist[v]==dist[u]+d[u][v]时，cnt[v]的值要再加上cnt[u]的值（v点的最短距离的路线多了一个u的路径）。

以上就是cnt[]数组的求法了。
此外还需要注意一个问题：在求cnt[v]时，必须要保证cnt[u]上的值已经为最终的结果了（即计算过程需要满足拓扑序）。因此，我们不能够用spfa算法来求解这道题。所以我们需要用bfs或dijkstra算法来求解这道题（bfs和dijkstra都是满足拓扑序的）。而这道题的边没有权值，因此我们可以用bfs。

代码如下

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <map>
#include <queue>
#include <vector>
#include <set>
#include <algorithm>
#define LL long long
#define ULL unsigned long long
#define PII pair<int,int>
#define x first
#define y second
using namespace std;
const int N=1e5+5,mod=100003,INF=0x3f3f3f3f;
vector<int> h[N];
int dist[N],cnt[N];
bool st[N];
void bfs()			//求最短路及方案数
{
    memset(dist,0x3f,sizeof dist);
    queue<int> q;
    q.push(1);
    dist[1]=0;
    cnt[1]=1;
    while(q.size())
    {
        int u=q.front();
        q.pop();
        
        if(st[u]) continue;
        st[u]=true;
        
        for(int i=0;i<h[u].size();i++)
        {
            int v=h[u][i];
            if(dist[v]>dist[u]+1)		//求解的两种情况见上面的分析
            {
                dist[v]=dist[u]+1;
                cnt[v]=cnt[u];
                q.push(v);
            }
            else if(dist[v]==dist[u]+1)
                cnt[v]=(cnt[v]+cnt[u])%mod;
        }
    }
}
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    while(m--)
    {
        int u,v;
        scanf("%d %d",&u,&v);
        h[u].push_back(v);
        h[v].push_back(u);
    }
    bfs();
    for(int i=1;i<=n;i++)
        printf("%d\n",cnt[i]);
    return 0;
}