洛谷 P1144 最短路计数

题目描述

给出一个NNN个顶点MMM条边的无向无权图，顶点编号为1−N1-N1−N。问从顶点111开始，到其他每个点的最短路有几条。
输入输出格式
输入格式：

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

接下来MMM行，每行222个正整数x,yx,yx,y，表示有一条顶点xxx连向顶点yyy的边，请注意可能有自环与重边。

输出格式：

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

输入输出样例

输入样例#1：

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

输出样例#1：

1
1
1
2
4

题意：一个单纯的无负环的无向图，然后告诉你哪两个点相互连接，并且距离为1，现在求1到达每个点的最短路径有多少条

解答：单纯的最短路条目数求解问题，num[maxn]表示1到每个节点的最短路径数数组，那么可以想到，num[1]=1，那么源点1到终点u有num[u]条最短路径，到终点v，而此时终点v的最短路径由dis[u]+mp[u][v]得到，那么源点1到达终点v的最短路径数目num[v]就得加上中间 的承接点u的最短路径数目，所以num[v]=num[v]+num[u];构成我们的递归方程，这个递归方程产生的数据量很大，所以题目要求取模，而num[v]=num[v]+num[u]的判断条件就是dis[v]=dis[u]+mp[u][v]时，如果dis[v]>dis[u]+mp[u][v]说明当前的不是最短路，最短路需要更新，那么num[v]原来的数据就不是最短路时的数据，需要更新，所以此时num[v]=num[u]，从新来过，num数组初始化都为0，然后源点num[1]=1;

这个题目数据量比较大，所以采用了迪杰斯特拉加最小堆优化来解决，还有邻接表储存数据。

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<queue>
#include<stack>
#include<string.h>
using namespace std;
const int N=1000010;
const int mod=100003;
struct Node{
	int v,next,w;
}edge[4*N];
int head[N],cnt;
void add_edge(int u,int v,int w)
{
	edge[cnt].v=v;
	edge[cnt].w=w;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
void init()
{
	memset(head,-1,sizeof(head));
	cnt=0;
}
int dis[N];
bool vis[N];
int n,m;
int num[N];
//建立最小堆优化
struct code{
	int u,dis;
	code(int u1,int dis1)
	{
		u=u1;
		dis=dis1;
	}
	bool operator<(const code &t)const
	{
		return dis>t.dis;
	}
};
priority_queue<code>q;
void Dij()
{
	for(int i=1;i<=n;i++)
	dis[i]=1e9,vis[i]=false;
	memset(num,0,sizeof(num));
	while(!q.empty())
	q.pop();
	dis[1]=0;
	num[1]=1;
	q.push(code(1,0));
	while(!q.empty())
	{
		code t=q.top();
		q.pop();
		if(vis[t.u])continue;
		vis[t.u]=true;
		for(int i=head[t.u];i!=-1;i=edge[i].next)
		{
			int v=edge[i].v;
			int w=edge[i].w;
			//如果路径距离相等，就加上num[u]
			if(dis[v]==dis[t.u]+w)
			{
				num[v]+=num[t.u];
				num[v]%=mod;
			}
			//更新最短路
			else if(dis[v]>dis[t.u]+w)
			{
				num[v]=num[t.u];
				dis[v]=dis[t.u]+w;
				q.push(code(v,dis[v]));
			}
		}
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	init();
	while(m--)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		add_edge(x,y,1);
		add_edge(y,x,1);
	}
	Dij();
	for(int i=1;i<=n;i++)
	printf("%d\n",num[i]);
}