【BFS求最短路】代码源每日一题div2 最短路计数

最短路计数 - 题目 - Daimayuan Online Judge

题意：

给出一个 N 个顶点 M 条边的无向无权图。

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

思路：

因为这是一个无权图，因此只需BFS求最短路即可

而它要求一个特定的起点到每个点的最短路有几条，只需要边BFS边递推即可

设结点1到结点i的最短路有ans[i]条，那么ans[i]就是周围所有结点的ans之和（dp的思想）

Code：

#include <bits/stdc++.h>
using namespace std;
const int mxn=1e6+10,mxe=2e6+10,mod=1e5+3;
queue<int> q;
struct ty{
    int to,next;
}edge[mxe<<1];
int n,m,x,y,cnt=0;
int head[mxn],dis[mxn],ans[mxn],vis[mxn];
void init(){
    for(int i=0;i<=n;i++){
        head[i]=-1;
        vis[i]=0;
    }
}
void add(int u,int v){
    edge[cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}
void bfs(){
    dis[1]=0;ans[1]=1;vis[1]=1;
    q.push(1);
    while(!q.empty()){
        int tmp=q.front();
        q.pop();
        for(int i=head[tmp];~i;i=edge[i].next){
            if(!vis[edge[i].to]){
                dis[edge[i].to]=dis[tmp]+1;
                vis[edge[i].to]=1;
                q.push(edge[i].to);
            }
            if(dis[edge[i].to]==dis[tmp]+1){
                ans[edge[i].to]=(ans[edge[i].to]+ans[tmp])%mod;
            }
        }
    }
}
int main(){
    scanf("%d%d",&n,&m);
    init();
    for(int i=1;i<=m;i++){
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    bfs();
    for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
    return 0;
}