BFS - 最短路计数 - 洛谷 P1144

BFS - 最短路计数 - 洛谷 P1144

给出一个 N 个顶点 M 条边的无向无权图，顶点编号为 1 到 N。

问从顶点 1 开始，到其他每个点的最短路有几条。

输入格式

第一行包含 2 个正整数 N,M，为图的顶点数与边数。

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

输出格式

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

如果无法到达顶点 i 则输出 0。

数据范围

$1\le N\le 10^5,1\le M\le 2\times 10^5$

输入样例：

5 7
1 2
1 3
2 4
3 4
2 3
4 5
4 5

输出样例：

1
1
1
2
4

---

分析：

$单源最短路，要统计源点到其它所有点的最短路径的条数。$

$距离数组dis[i]表示源点到i的最短距离，计数数组cnt[i]表示从源点到i点的最短路径条数。$

$因为是无向无权图，可以用bfs来求最短路径，时间复杂度是线性的。$

$记当前点的编号为t，与之相连通的点是j，分两种情况：$

$①、经过点t再到j的距离要更短：dis[j]>dis[t]+1，则$

$更新到j的最短路径：dis[j]=dis[t]+1，cnt[j]=cnt[t]。$

$②、经过点t再到j的距离与经过其他点到j的最短距离相同：dis[j]=dis[t]+1，则$

$将经过点t再到j的路径条数累加到经过其他点到j的最短路径上：cnt[j]+=cnt[t]。$

注意：

$无向边注意数组要开两倍。$

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>

using namespace std;

const int N=1e5+10, M=4e5+10, mod=100003;

int n,m;
int e[M],ne[M],h[N],idx;
int dis[N],cnt[N];

void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void bfs()
{
    queue<int> Q;
    
    memset(dis,0x3f,sizeof dis);
    dis[1]=0;
    cnt[1]=1;
    Q.push(1);
    
    while(Q.size())
    {
        int t=Q.front();
        Q.pop();
        
        for(int i=h[t];~i;i=ne[i])
        {
            int j=e[i];
            if(dis[j]>dis[t]+1)
            {
                dis[j]=dis[t]+1;
                cnt[j]=cnt[t];
                Q.push(j);
            }
            else if(dis[j]==dis[t]+1)
            {
                cnt[j]=(cnt[j]+cnt[t])%mod;
            }
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    
    memset(h,-1,sizeof h);
    int a,b;
    while(m--)
    {
        scanf("%d%d",&a,&b);
        add(a,b),add(b,a);
    }
    
    bfs();
    
    for(int i=1;i<=n;i++) printf("%d\n",cnt[i]);
    
    return 0;
}