最短路计数(bfs)

https://www.luogu.com.cn/problem/P1144

  • 邻接矩阵存图
  • 从1出发bfs
  • 同时记录深度
  • 如果下一个点的深度是起点深度+1,那么此刻应该累加路径的条数。
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define MAX 1000001
vector<int> G[MAX];
int n, m, x, y;
void input()
{
    cin >> n >> m;
    for (int i = 1; i <= m; i++)
    {
        scanf("%d%d", &x, &y);
        G[x].push_back(y);
        G[y].push_back(x);
    }
}
int seen[MAX];
int ans[MAX];
int depth[MAX];
#define MOD 100003
void bfs()
{
    queue<int> q;
    q.push(1);
    ans[1] = 1;
    seen[1] = 1;
    while (!q.empty())
    {
        int now = q.front();
        q.pop();
        for (int i = 0; i < G[now].size(); i++)
        {
            int to = G[now][i];
            if (!seen[to])
            {
                seen[to] = 1;
                q.push(to);
                depth[to] = depth[now] + 1;
            }
            if (depth[to] == depth[now] + 1)
            {
                ans[to] = (ans[to] + ans[now]) % MOD;
            }
        }
    }
    for (int i = 1; i <= n; i++)
    {
        printf("%d\n", ans[i]);
    }
}
int main()
{
    input();
    bfs();
}

 

posted on 2022-07-30 23:42  樵风  阅读(55)  评论(0编辑  收藏  举报