neuq-acm预备队训练week 8 P1144 最短路计数
题目描述
给出一个 N 个顶点 M条边的无向无权图，顶点编号为 1∼N。问从顶点 1 开始，到其他每个点的最短路有几条。
题目限制
输入格式
第一行包含 22 个正整数 N,M，为图的顶点数与边数。
接下来 M 行，每行 2个正整数 x,y，表示有一条由顶点 x 连向顶点 y 的边，请注意可能有自环与重边。
输出格式
共 N 行，每行一个非负整数，第 i 行输出从顶点 1 到顶点 i 有多少条不同的最短路，由于答案有可能会很大，你只需要输出 ans mod 100003 后的结果即可。如果无法到达顶点 i 则输出 0。
输入输出样例
解题思路
基于图论的DFS，我们可以通过 记忆化搜索 或者 DP 进行优化。我们这里选择使用DP。用SPFA预处理跑一边最短路
AC代码
#include <bits/stdc++.h>
#define inf 0x7FFFFFFF
using namespace std;
#define Max 2000005
#define mod 100003
int n,m,head[Max],cnt,dis[Max],dp[Max];
bool inq[Max],vis[Max];struct edge{int to,nxt;
}e[Max];
void SPFA();
void DP();
void add(int u,int v);
int main()
{cin>>n>>m;for(int i=1;i<=m;++i){int a,b;cin>>a>>b;add(a,b),add(b,a);}SPFA();DP();for(int i=1;i<=n;++i) printf("%d\n",dp[i]);return 0;
}void add(int u,int v)
{e[++cnt].to = v;e[cnt].nxt = head[u];head[u] = cnt;
}void SPFA()
{priority_queue <pair<int,int> > q;for(int i=2 ;i<=n;++i) dis[i] = inf;q.push(make_pair(0,1));inq[1] = 1;while(!q.empty()){int now = q.top().second;q.pop();inq[now] = 0;for(int i = head[now];i;i = e[i].nxt){int to = e[i].to;if(dis[to] > dis[now] + 1){dis[to] = dis[now] + 1;q.push(make_pair(-dis[to],to));inq[to] = 1;}}}
}
void DP()
{memset(vis,0,sizeof(vis));queue <int> q;q.push(1);dp[1]=1;vis[1]=1;dis[1]=0;while(!q.empty()){int now = q.front();q.pop();for(int i=head[now];i;i=e[i].nxt){int to = e[i].to;if(dis[to] == dis[now] + 1){dp[to] += dp[now];dp[to] %= mod;if(!vis[to])q.push(to),vis[to] = 1;}}}
}