D - Number of Shortest paths (最短路径的条数 BFS)

春春飞舞耶

已于 2024-06-02 14:56:17 修改

阅读量403

点赞数 1

分类专栏： AtCoder 文章标签：宽度优先算法

于 2021-08-11 21:28:16 首次发布

本文链接：https://blog.csdn.net/qq_59773176/article/details/119617433

版权

AtCoder 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

该博客主要介绍了如何利用广度优先搜索（BFS）算法解决图论问题，具体是计算从源点到所有其他顶点的最短路径的条数。在给定的无权图中，每条边的权重被设为1，通过BFS遍历更新节点的距离和最短路径计数，最终得到最短路径总数。

摘要由CSDN通过智能技术生成

题目链接
在这里插入图片描述

每条边的边权都是1，对于每个从 v 有一条边的 u ，如果我们还没有访问过 u ，将 dis [ u ] 更新为 dis [ v ] + 1 并将 cnt [ u ] 加上 cnt [ v ] ；如果我们已经访问过 u 并且 dis [ u ] = dis [ v ] + 1 ，则只将 cnt [ u ] 加上 cnt [ v ]即可；否则，跳过。
最后cnt [ n ] 即是最短路条数

#include <bits/stdc++.h>
using namespace std;
#define mem(a, b) memset(a, b, sizeof(a))
#define ll long long int
#define endl '\n'
typedef pair<int, int> PII;
const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
const int N = 1e6 + 7;
 
int Head[N], Edge[N], Next[N], idx;
void add(int a, int b) {
    idx++;
    Edge[idx] = b;
    Next[idx] = Head[a];
    Head[a] = idx;
}
 
int n, m;
int dis[N], cnt[N];
int vis[N];
void bfs(int s) {
    queue<int> q;
    dis[s] = 0;
    q.push(s);
    vis[s] = 1;
    cnt[s] = 1;
    while (!q.empty()) {
        int t = q.front();
        q.pop();
        for (int i = Head[t]; ~i; i = Next[i]) {
            int e = Edge[i];
            if (!vis[e]) {
                dis[e] = dis[t] + 1;
                cnt[e] = cnt[t];
                q.push(e);
                vis[e] = 1;
            } else if (vis[e] && dis[e] == dis[t] + 1)
                cnt[e] = (cnt[e] + cnt[t]) % mod;
        }
    }
}
signed main() {
    mem(Head, -1);
    cin >> n >> m;
    for (int i = 1; i <= m; i++) {
        int u, v;
        cin >> u >> v;
        add(u, v);
        add(v, u);
    }
    bfs(1);
    cout << cnt[n] << endl;
    return 0;
}