单源最短路的扩展应用---------最短路计数

给出一个 NN 个顶点 MM 条边的无向无权图,顶点编号为 11 到 NN。
问从顶点 11 开始,到其他每个点的最短路有几条。
输入格式
第一行包含 22 个正整数 N,MN,M,为图的顶点数与边数。
接下来 MM 行,每行两个正整数 x,yx,y,表示有一条顶点 xx 连向顶点 yy 的边,请注意可能有自环与重边。
输出格式
输出 NN 行,每行一个非负整数,第 ii 行输出从顶点 11 到顶点 ii 有多少条不同的最短路,由于答案有可能会很大,你只需要输出对 100003100003 取模后的结果即可。
如果无法到达顶点 ii 则输出 00。
数据范围
1≤N≤1051≤N≤105,

1≤M≤2×1051≤M≤2×105
输入样例:
5 7
1 2
1 3
2 4
3 4
2 3
4 5
4 5

输出样例:
1
1
1
2
4

#include 
#include 
#include 
#include 
using namespace std;
const int N = 100010, M = 400010, mod = 100003;
int n, m;
int h[N], e[M], ne[M], idx;
int dist[N], cnt[N];
int q[N];
void add(int a, int b){
 e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
void bfs(){
 memset(dist, 0x3f, sizeof dist);
 int hh = 0, tt = 0;
  q[0] = 1;
 dist[1] = 0;
 cnt[1] = 1;
  while(hh <= tt){
  int t = q[hh ++];
  for (int i = h[t]; ~i; i = ne[i]){
   int j = e[i];
   if (dist[j] > dist[t] + 1){
    dist[j] = dist[t] + 1;
    cnt[j] = cnt[t];
    q[++ tt] = j;
   }
   else if (dist[j] == dist[t] + 1)
     cnt[j] = (cnt[j] + cnt[t]) % mod;
  }
 }
}
int main(){
 scanf("%d%d", &n, &m);
 memset(h, -1, sizeof h);
  while(m --){
  int a, b;
  scanf("%d%d", &a, &b);
  add(a, b), add(b, a);
 }
  bfs();
  for (int i = 1; i <= n; i ++)   cout << cnt[i] << endl;
  return 0;
}

你可能感兴趣的:(c++,图论,算法,深度优先)