洛谷 P1144 最短路计数

题目描述

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

输入输出格式

输入格式：

 

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

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

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 复制

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

输出样例#1： 复制

1
1
1
2
4

说明

11到55的最短路有44条，分别为22条1-2-4-51−2−4−5和22条1-3-4-51−3−4−5（由于4-54−5的边有22条）。

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

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

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

首先可以考虑自环对这个题是没有影响的，重边是有的 ， 稍微处理一下，因为边权为1，所以直接bfs，类似spfa标记数组，第一次出现 入队，并且记录计数，后面再出现，不用再入队，但是要统计当前结果对答案做的贡献，就是当v到u的时候，dis[v] + 1 == dis[u]

#include <bits/stdc++.h>
#include <time.h>
#define fi first
#define se second
#define endll "\n"
#define MS0(X) memset((X), 0, sizeof((X)))
#define MS1(X) memset((X), -1, sizeof((X)))
#define LEN(X) strlen(X)
///vector(len,val);
using namespace std;

typedef long long ll;
typedef double db;
int xx[4] = {1,-1,0,0};
int yy[4] = {0,0,1,-1};
const double eps = 1e-9;
typedef pair<int,int>  P;
const int maxn = 2e6 + 5000;
const ll mod = 1e9 + 7;
inline int sign(db a) { return a < -eps ? -1 : a > eps;}
inline int cmp(db a,db b){ return sign(a - b);}
void debug(int x){  cout << x << endl; }
ll mul(ll a,ll b,ll c) { ll res = 1; while(b) {  if(b & 1) res *= a,res %= c;  a *= a,a %= c,b >>= 1;  }  return res;}
ll phi(ll x) {  ll res = x;  for(ll i = 2; i * i <= x; i++) { if(x % i == 0) res = res / i * (i - 1);   while(x % i == 0) x /= i;   }  if(x > 1) res = res / x  * (x - 1);    return res;}

int fa[maxn];
int Find(int x) {   if(x != fa[x]) return fa[x] = Find(fa[x]);  return fa[x];}
ll c,n,k; ll dis[maxn];
bool vis[maxn];
const ll modd = 100003;
ll ans[maxn];
vector<int>v[maxn];
void bfs()
{
    int st = 1;
    memset(vis,0,sizeof(vis));
    queue<int>q;
    q.push(st);
    dis[st] = 0;
    vis[st] = 1;
    ans[st] = 1;
    while(!q.empty()){
        int x = q.front(); q.pop();
        for(auto d:v[x]){
            if(!vis[d]){
                if(dis[d] > dis[x] + 1){
                    dis[d] = dis[x] + 1;
                    ans[d] += ans[x],ans[d] %= modd;
                    if(!vis[d]) q.push(d);
                    vis[d] = 1;
                }
            }else{
                if(dis[d] == dis[x] + 1)  ans[d] += ans[x],ans[d] %= modd;
            }
        }
    }
    return ;
}
int main() { int m;
    ios::sync_with_stdio(false);
    while(cin >> n >> m){
        for(int i = 1;i <= n;i++) dis[i] = 1e18;
        for(int i = 1;i <= m;i++){
         int x,y;
         cin >> x >> y;
         if(x == y) continue;
         v[x].push_back(y);
         v[y].push_back(x);
        }
        bfs();
        for(int i = 1;i <= n;i++){
            cout << ans[i] << endl;
        }
    }
    cerr << "time: " << (long long)clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
    return 0;
}