P1144 最短路计数

题目描述

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

输入格式

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

接下来 M 行，每行 2 个正整数 x,y，表示有一条连接顶点 x 和顶点 y 的边，请注意可能有自环与重边。

输出格式

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

输入输出样例

输入 #1复制

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

输出 #1复制

1
1
1
2
4

说明/提示

1 到 5 的最短路有 4 条，分别为 2 条 1→2→4→5 和 2 条 1→3→4→5（由于 4→5 的边有 2 条）。

对于 20% 的数据，1≤N≤100；
对于 60% 的数据，1≤N≤103；
对于 100% 的数据，1≤N≤106，1≤M≤2×106。

做了一下午终于A了这题，谈谈心得。

这题主要思路是这样的，在SPFA中加入计数操作，如果这个点的最短路可以更新但还未被更新既

if(dis[v]>dis[u]+1)

那么很显然，目前我们可以知道的可到达这个点的最短路就是通过先到达上个点再走当前边的方式，易推得代码

ans[v]=ans[u];

当一个点不能被当前点更新时，判断如果这个点的距离等于当前点距离+1的话，那么就如同上面的情况，可到达这个点的最短路有一种是通过先到达上个点再走当前边的方式得到的，那么我们就把这个情况加上，可得代码

ans[v]+=ans[u]

将这个计数方法放到SPFA的模板中就可以ac本题

---

几个注意事项

1.边加边膜 防止爆int

2.ans[1]=1。因为很显然，从一到一的最短路有且只有一条、

3.存双向边

4.自环和重边并不会对本题造成影响

贴上ac代码

#include<bits/stdc++.h>
using namespace std;
const int mod=100003;
const int Maxn=1000010;
const int MAxn=5000010;
const int MAXn=0x7fffffff;
struct st{
	int to,nxt;
}edge[MAxn];
int n,m,cnt=0,dis[Maxn];
int head[Maxn],vis[Maxn],ans[Maxn];
inline void add_edge(int u,int v){
	edge[++cnt].nxt=head[u];
	edge[cnt].to=v;
	head[u]=cnt;
}
inline void SPFA(int s){
	for(int i=1;i<=n;++i) dis[i]=MAXn;
	dis[s]=0; vis[s]=1;
	queue<int> Q;
	Q.push(s);
	while(!Q.empty()){
		int u=Q.front(); 
		Q.pop();vis[u]=0;
		for(int i=head[u];i;i=edge[i].nxt){
			int v=edge[i].to;
			if(dis[v]>dis[u]+1){
				dis[v]=dis[u]+1;
				ans[v]=ans[u];
                ans[v]%=mod;
				if(!vis[v]){
					vis[v]=1;
					Q.push(v);
				}
			}
			else if(dis[v]==dis[u]+1){
				ans[v]+=ans[u];
				ans[v]%=mod;
			}
		}
	}
} 
int main(){
	ans[1]=1;
	memset(vis,0,sizeof(vis));
	scanf("%d %d",&n,&m);
	for(int i=1;i<=m;++i){
		int x,y;
		scanf("%d %d",&x,&y);
		add_edge(x,y);
		add_edge(y,x);
	}
	SPFA(1);
	for(int i=1;i<=n;++i) printf("%d\n",ans[i]);
	return 0;
}