1. 程式人生 > >【題解】洛谷P1144最短路計數 spfa

【題解】洛谷P1144最短路計數 spfa

題目連結

在這裡插入圖片描述 在這裡插入圖片描述

spfa跑單源最短路,注意相等時線路數相加。

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int N=1e6+10,mod=1e5+3;
int n,m,hd[N],tot,dis[N],vis[N],cnt[N];
struct Edge{
	int v,nx;
}e[N<<2];
inline void add(int u,int v)
{
	e[tot].v=v;
	e[tot].nx=hd[u];
	hd[u]=tot++;
}
inline void spfa()
{
	memset(dis,0x3f,sizeof(dis));
	queue<int>q;dis[1]=0;vis[1]=1;q.push(1);cnt[1]=1;
	while(q.size())
	{
		int u=q.front();q.pop();vis[u]=0;
		for(int i=hd[u];~i;i=e[i].nx)
		{
			int v=e[i].v;
			if(dis[v]>dis[u]+1)
			{
				dis[v]=dis[u]+1;cnt[v]=cnt[u];
				if(!vis[v])q.push(v),vis[v]=1;
			}
			else if(dis[v]==dis[u]+1)
			{
				cnt[v]+=cnt[u];if(cnt[v]>=mod)cnt[v]-=mod;
			}
		}
	}
}
int main()
{
	//freopen("in.txt","r",stdin);
    memset(hd,-1,sizeof(hd));
	scanf("%d%d",&n,&m);
    int u,v;
    for(int i=1;i<=m;i++)
        scanf("%d%d",&u,&v),add(u,v),add(v,u);
    spfa();
    for(int i=1;i<=n;i++)
        printf("%d\n",cnt[i]);
    return 0;
}

總結

做逛公園的前置技能