【題解】洛谷P1144最短路計數 spfa
阿新 • • 發佈:2018-12-14
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; }
總結
做逛公園的前置技能