题目描述
给出一个 N 个顶点 M 条边的无向无权图,顶点编号为 1 到 N。
问从顶点 1 开始,到其他每个点的最短路有几条。
输入格式
第一行包含 2 个正整数 N,M,为图的顶点数与边数。
接下来 M 行,每行两个正整数 x,y,表示有一条顶点 x 连向顶点 y 的边,请注意可能有自环与重边。
输出格式
输出 N 行,每行一个非负整数,第 i 行输出从顶点 1 到顶点 i 有多少条不同的最短路,由于答案有可能会很大,你只需要输出对 100003 取模后的结果即可。
如果无法到达顶点 i 则输出 0。
数据范围
1≤N≤105,
1≤M≤2×105
输入样例
5 7
1 2
1 3
2 4
3 4
2 3
4 5
4 5
输出样例
1
1
1
2
4
题目分析
因为要求的是方案数,因此我们在求最短路中的dist[]时,还要再额外维护一个cnt[]数组来记录最短路的方案数。
- 当dist[v]>dist[u]+d[u][v]时,cnt[v]的值即为cnt[u]的值(因为v点最短距离是由u点的最短距离更新而来)。
- dist[v]==dist[u]+d[u][v]时,cnt[v]的值要再加上cnt[u]的值(v点的最短距离的路线多了一个u的路径)。
以上就是cnt[]数组的求法了。
此外还需要注意一个问题:在求cnt[v]时,必须要保证cnt[u]上的值已经为最终的结果了(即计算过程需要满足拓扑序)。
因此,我们不能够用spfa算法来求解这道题。所以我们需要用bfs或dijkstra算法来求解这道题(bfs和dijkstra都是满足拓扑序的)。而这道题的边没有权值,因此我们可以用bfs。
代码如下
#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <map>
#include <queue>
#include <vector>
#include <set>
#include <algorithm>
#define LL long long
#define ULL unsigned long long
#define PII pair<int,int>
#define x first
#define y second
using namespace std;
const int N=1e5+5,mod=100003,INF=0x3f3f3f3f;
vector<int> h[N];
int dist[N],cnt[N];
bool st[N];
void bfs() //求最短路及方案数
{
memset(dist,0x3f,sizeof dist);
queue<int> q;
q.push(1);
dist[1]=0;
cnt[1]=1;
while(q.size())
{
int u=q.front();
q.pop();
if(st[u]) continue;
st[u]=true;
for(int i=0;i<h[u].size();i++)
{
int v=h[u][i];
if(dist[v]>dist[u]+1) //求解的两种情况见上面的分析
{
dist[v]=dist[u]+1;
cnt[v]=cnt[u];
q.push(v);
}
else if(dist[v]==dist[u]+1)
cnt[v]=(cnt[v]+cnt[u])%mod;
}
}
}
int main()
{
int n,m;
scanf("%d %d",&n,&m);
while(m--)
{
int u,v;
scanf("%d %d",&u,&v);
h[u].push_back(v);
h[v].push_back(u);
}
bfs();
for(int i=1;i<=n;i++)
printf("%d\n",cnt[i]);
return 0;
}