洛谷最短路计数

最新推荐文章于 2023-08-04 08:11:52 发布

權权诠

最新推荐文章于 2023-08-04 08:11:52 发布

阅读量150

点赞数 1

分类专栏：洛谷最短路题单文章标签：图论算法数据结构

本文链接：https://blog.csdn.net/WULULQ/article/details/125143457

版权

最短路 BFS算法无向无权图路径计数广度优先搜索

关键词由CSDN通过智能技术生成

洛谷最短路题单专栏收录该内容

3 篇文章 0 订阅

订阅专栏

最短路计数

题目描述

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

输入格式

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

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

输出格式

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

样例 #1

样例输入 #1

样例输出 #1

提示

$1$ 到 $5$ 的最短路有 $4$ 条，分别为 $2$ 条 $1 - 2 - 4 - 5$ 和 $2$ 条 $1 - 3 - 4 - 5$ （由于 $4 - 5$ 的边有 $2$ 条）。

对于 $20\%$ 的数据， $N \leq 100$ ；

对于 $60\%$ 的数据， $N \leq 1000$ ；

对于 $100\%$ 的数据， $N < = 1000000, M < = 2000000$ 。

思路

距离各点相同，我们很轻易可以想到用bfs来求最短路，这里要注意的一个点就是，1点到各点的距离可以表示为各点在bfs搜索树中的深度。
那么我们在bfs过程中最重要的思路就是：
遍历一个点，如果是第一次访问，那么记录该点，ans[j]=ans[t],取模，标记深度,将j入队，用j来更新其他剩余点
如果不是第一次访问，则进行判断，如果深度相同，则是一条相同距离的最短路，ans+=ans[t]，取模。

题解

#include <iostream>
#include <queue>
#include <cstring>
using namespace std;
const int N = 1e6 + 10;
const int M = 2e6 + 10;
const int mod = 100003;
int e[M], ne[M], h[N], idx;
int dis[N], visit[N], ans[N]; // dis表示深度，ans记录距离

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
void bfs()
{ //获取从1点到所有点的最短路
    int s = 1;
    queue<int> q;
    q.push(s);
    visit[1] = 1;
    ans[1] = 1;
    while (q.size())
    {
        int t = q.front();
        q.pop();
        for (int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            if (!visit[j]) //如果该点未访问，则深度等于前一个点加一，ans等于前一个点
            {
                visit[j] = 1;
                q.push(j); //只在第一次访问的时候加入队列，否则会重复遍历点
                dis[j] = dis[t] + 1;
                ans[j] = ans[t];
                ans[j] %= mod;
            }
            else //如果该点已访问，则ans+=前一个点
            {
                if (dis[j] == dis[t] + 1)
                {
                    ans[j] += ans[t];
                    ans[j] %= mod;
                }
            }
        }
    }
}
int main()
{
    memset(h, -1, sizeof h);
    int n, m;
    cin >> n >> m;
    for (int i = 0; i < m; i++)
    {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b);
        add(b, a);
    }
    bfs();
    for (int i = 1; i <= n; i++)
        printf("%d\n", ans[i]);
    return 0;
}