P1144 最短路计数-dijkstra+bfs

最短路计数

题目描述

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

输入格式

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

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

输出格式

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

样例 #1

样例输入 #1

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

样例输出 #1

1
1
1
2
4

提示

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

对于 $20\%$ 的数据，$1\le N\le 100$；
对于 $60\%$ 的数据，$1\le N\le 10^3$；
对于 $100\%$ 的数据，$1\le N\le 10^6$，$1\le M\le 2\times 10^6$。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define de(x) cout<<x<<" ";
#define sf(x) scanf("%d",&x);
#define Pu puts("");
const int N=1e6+10,M=2e6+10,mod=100003;
int hed[N],nxt[M<<1],var[M<<1],tot;
int n,m;
int cnt[N],d[N];
int vis[N];
void _add(int x,int y){
    var[++tot]=y;
    nxt[tot]=hed[x];
    hed[x]=tot;
}
void fun(){
    memset(d,0x3f,sizeof(d));
    cnt[1]=1;d[1]=0;
    queue<int>q;
    q.push(1);
    int w,t;
    while(!q.empty()){
        w=q.front();q.pop();
        if(vis[w]) continue;
        vis[w]++;
        //其实这个地方放在for循环里面也是一样的
        //由于权重为1，不存在求《最短路时的问题,vis数组更新位置的问题》
        for(int i=hed[w];i;i=nxt[i]){
            t=var[i];
            if(d[t]>d[w]+1){
                d[t]=d[w]+1;
                //注意这里：是在前一个节点w的基础上，继续向下
                cnt[t]=cnt[w]%mod;
                q.push(t);
            }else if(d[t]==d[w]+1){
                //注意这里：是自身节点t和上一个节点的和
                cnt[t]=(cnt[t]+cnt[w])%mod;
            }
        }
    }
}
int main(){
    cin>>n>>m;
    int x,y;
    for(int i=1;i<=m;i++){
        sf(x)sf(y)
        _add(x,y);
        _add(y,x);
        //vis[x]++;
        //原本想这样处理vis数组，但是后来发现一个问题
        //就是比如1连接2和3,当处理完1后，它任然可以继续遍历
        //但是这个时候，cnt[2]就会被再次加一，结果显然是错误的
    }
    fun();
    for(int i=1;i<=n;i++){
        printf("%d\n",cnt[i]);
    }
    return 0;
}