首页 > 其他 > 详细

最短路计数

时间:2021-05-23 22:41:44      阅读:21      评论:0      收藏:0      [点我收藏+]

将计数融入最短路中,即

if(dis[v]==dis[u]+edge[i].w){
  ans[v]+=ans[u];
  ans[v]%=mod;
}
#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int N=1000005;
const int mod=100003;
int n,m,tot;
int head[N],ans[N],dis[N];
bool vis[N];
struct node{
	int to,next;
}edge[N<<1];
void add(int u,int v){
	edge[tot].to=v;
	edge[tot].next=head[u];
	head[u]=tot++;
}
void dij(int x){
	ans[x]=1;
	priority_queue<pair<int,int> > q;
	dis[x]=0; q.push(make_pair(0,x));
	while(!q.empty()){
		int u=q.top().second; q.pop();
		if(vis[u]) continue; vis[u]=1;
		for(int i=head[u];i!=-1;i=edge[i].next){
			int v=edge[i].to;
			if(dis[v]>dis[u]+1){
				dis[v]=dis[u]+1;
				q.push(make_pair(-dis[v],v));
			}
			if(dis[v]==dis[u]+1) ans[v]+=ans[u],ans[v]%=mod;
		}
	}
}
int main(){
	memset(head,-1,sizeof(head));
	memset(dis,0x3f3f3f,sizeof(dis));
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++){
		int x,y;
		scanf("%d%d",&x,&y);
		add(x,y); add(y,x);
	}
	dij(1);
	for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
	return 0;
}

最短路计数

原文:https://www.cnblogs.com/New-ljx/p/14801501.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!