最短路计数
提交数: 51, 通过率: 54.9%, 平均分: 65.49
题目描述:
给出一个N个顶点M条边的无向无权图,顶点编号为1~N。问从顶点1开始,到其他每个点的最短路有几条?
输入格式:
第一行包含2个正整数N, M, 分别为图的顶点数与边数。
接下来M行,每行两个正整数x, y,表示有一条从顶点x连向点y的边。请注意可能有自环与重边。
输出格式:
共N行,每行一个非负整数,第i行输出从顶点1到顶点i有多少条不同的最短路,由于结果的值有可能会很大,你只需要输出结果 mod 100003即可。如果无法到达顶点i,则输出0。
数据范围:
对于20%的数据,N<=100。
对于60%的数据,N<=1000。
对于100%的数据,N<=100,000, M<=200,000。
样例输入:
5 7
1 2
1 3
2 4
3 4
2 3
4 5
4 5
样例输出:
1
1
1
2
4
提示:
样例解释:
1到5的最短路有4条,分别为2条1-2-4-5和2条1-3-4-5 ( 由于4-5的边有2条)。
此题使用SPFA(Shortest Path Faster Algorithm)
#include <bits/stdc++.h>
using namespace std;
int n,m,f[1000005],t[4000005],n1[4000005],g,dis[1000005],ans[1000005];
bool book[1000005];
queue<int> que;
void SPFA(int u,int v) {
t[g]=v,n1[g]=f[u],f[u]=g++;
}
int dfs(int u) {
if(ans[u])return ans[u];//如果ans[u]!=0直接输出,否则更新并输出
for(int i=f[u]; i!=-1; i=n1[i])
if(dis[u]-1==dis[t[i]])ans[u]=(ans[u]+dfs(t[i]))%100003;//统计最短路的条数并更新
return ans[u];
}
int main() {
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++)f[i]=-1,dis[i]=INT_MAX;//将数组初始化
for(int i=1,u,v; i<=m; i++)scanf("%d%d",&u,&v),SPFA(u,v),SPFA(v,u);//输入之后构建邻接矩阵
dis[1]=0,que.push(1),book[1]=1;
while(!que.empty()) {//循环寻找最短路
int f1=que.front();//更新出发点
que.pop(),book[f1]=0;//删除已遍历的元素
for(int i=f[f1]; i!=-1; i=n1[i])//遍历出发的边
if(dis[f1]+1<dis[t[i]]) {//如果比之前的最短路短
dis[t[i]]=dis[f1]+1;//更新最短路
if(!book[t[i]])que.push(t[i]),book[t[i]]=1;//标记已遍历过的节点和边
}
}
ans[1]=1;
for(int i=1; i<=n; i++)printf("%d\n",dfs(i));//输出最短路的个数
return 0;
}