洛谷2505 [HAOI2012]道路(最短路计数)
生活随笔
收集整理的這篇文章主要介紹了
洛谷2505 [HAOI2012]道路(最短路计数)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
洛谷傳送門
【題目分析】
線段樹?bczd,這么小的范圍直接暴力就行啦。
直接O(n)枚舉源點,每次跑最短路,然后對于每一條路徑統計是否在最短路上、兩個端點各有多少條最短路徑經過即可。
【代碼~】
#include<bits/stdc++.h> using namespace std; const int MAXN=1e3+510; const int MAXM=1e4+10; const int INF=0x3f3f3f3f; const int MOD=1e9+7;int n,m,cnt; int head[MAXN],dis[MAXN],vis[MAXN]; int nxt[MAXM],from[MAXM],to[MAXM],w[MAXM]; int dp1[MAXN],dp2[MAXN]; int du[MAXN]; int ans[MAXM];int Read(){int i=0,f=1;char c;for(c=getchar();(c>'9'||c<'0')&&c!='-';c=getchar());if(c=='-')f=-1,c=getchar();for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';return i*f; }void add(int x,int y,int z){nxt[cnt]=head[x];head[x]=cnt;from[cnt]=x;to[cnt]=y;w[cnt]=z;cnt++; }void SPFA(int s){queue<int> q;memset(dis,INF,sizeof(dis));memset(vis,0,sizeof(vis));memset(du,0,sizeof(du));memset(dp1,0,sizeof(dp1));memset(dp2,0,sizeof(dp2));dis[s]=0;vis[s]=1;q.push(s);while(!q.empty()){int u=q.front();q.pop();vis[u]=0;for(int i=head[u];i!=-1;i=nxt[i]){int v=to[i];if(dis[v]>dis[u]+w[i]){dis[v]=dis[u]+w[i];if(!vis[v]){vis[v]=1;q.push(v);}}}}for(int i=0;i<cnt;++i){int u=from[i],v=to[i];if(dis[v]==dis[u]+w[i]){du[v]++;}} }void bfs(int s){queue<int> q;q.push(s);dp1[s]=1;while(!q.empty()){int u=q.front();q.pop();for(int i=head[u];i!=-1;i=nxt[i]){int v=to[i];if(dis[v]==dis[u]+w[i]){dp1[v]=(dp1[v]+dp1[u])%MOD;if(!--du[v]){q.push(v);}}}} }int dfs(int u){if(dp2[u])return dp2[u];dp2[u]=1;for(int i=head[u];i!=-1;i=nxt[i]){int v=to[i];if(dis[v]==dis[u]+w[i]){dp2[u]=(dp2[u]+dfs(v))%MOD;}}return dp2[u]; }int main(){memset(head,-1,sizeof(head));n=Read(),m=Read();for(int i=1;i<=m;++i){int x=Read(),y=Read(),z=Read();add(x,y,z);}for(int i=1;i<=n;++i){SPFA(i);bfs(i);dfs(i);for(int j=0;j<cnt;++j){int u=from[j],v=to[j];if(dis[u]+w[j]==dis[v]){ans[j]=(ans[j]+(long long)(dp1[u]*dp2[v])%MOD)%MOD;}}}for(int i=0;i<cnt;++i)cout<<ans[i]<<'\n';return 0; }?
轉載于:https://www.cnblogs.com/Ishtar/p/10291868.html
總結
以上是生活随笔為你收集整理的洛谷2505 [HAOI2012]道路(最短路计数)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: js 为false的几种情况
- 下一篇: 微信授权(Net Mvc)