1. 程式人生 > >BZOJ 2750: [HAOI2012]Road

BZOJ 2750: [HAOI2012]Road

其實也就是個最短路。。。

以每個點為起點 都跑一次最短路 然後統計一下就好了。。
對於在最短路圖上的邊(x,y)貢獻就是走到x的方案*y有多少種走法走到任意一點
可能比較拗口,隨便yy一下就好。。

顯然是個有向無環圖,正著反著都拓撲一下就能算出來了。。
為了偷懶,我直接按dis排序了,邊權>0 這樣顯然滿足拓撲序(不過這樣好像就會慢一點…

一開始還因為SB錯誤WA了一發。。。

#include<bits/stdc++.h>
using namespace std;
const int N=1502,M=5002,mod=1e9+7;
char O[1<<14
],*S=O,*T=O; #define gc (S==T&&(T=(S=O)+fread(O,1,1<<14,stdin),S==T)?-1:*S++) inline int read(){ int x=0,f=1; char ch=gc; while(ch<'0' || ch>'9'){if(ch=='-')f=-1; ch=gc;} while(ch>='0' && ch<='9'){x=(x<<1)+(x<<3)+(ch^48); ch=gc;} return x*f; } struct
edg{int y,c,nex;}e1[M],e2[M]; int f1[N],f2[N],len; void ins(int x,int y,int c){ e1[++len]=(edg){y,c,f1[x]},f1[x]=len; e2[ len]=(edg){x,c,f2[y]},f2[y]=len; } int ans[M],fr[N],to[N],d[N],n,a[N]; queue<int>q; bool v[N]; void spfa(int x){ for(int i=1;i<=n;++i) d[i]=mod,v[i]=0; q.push(x); v[x]=1
,d[x]=0; while(!q.empty()){ x=q.front(); q.pop(); v[x]=0; for(int k=f1[x];k;k=e1[k].nex){ int y=e1[k].y; if(d[y]>d[x]+e1[k].c){ d[y]=d[x]+e1[k].c; if(!v[y]) v[y]=1,q.push(y); } } } } int Cmp1(int x,int y){return d[x]<d[y];} int Cmp2(int x,int y){return d[x]>d[y];} void solve(int st){ spfa(st); int i; for(i=1;i<=n;++i)fr[i]=0,to[i]=1; sort(a+1,a+1+n,Cmp1); fr[st]=1; for(i=1;i<=n;++i){ int x=a[i]; for(int k=f1[x];k;k=e1[k].nex){ int y=e1[k].y; if(d[y]==d[x]+e1[k].c)fr[y]+=fr[x]; } } sort(a+1,a+1+n,Cmp2); for(i=1;i<=n;++i){ int x=a[i]; for(int k=f2[x];k;k=e2[k].nex){ int y=e2[k].y; if(d[y]+e2[k].c==d[x])to[y]+=to[x]; } } for(int x=1;x<=n;++x) for(int k=f1[x];k;k=e1[k].nex){ int y=e1[k].y; if(d[y]==d[x]+e1[k].c) ans[k]=(ans[k]+1ll*fr[x]*to[y]%mod)%mod; } } int main(){ n=read(); int i,m=read(); for(i=1;i<=m;++i){ int x=read(),y=read(),c=read(); ins(x,y,c); } for(i=1;i<=n;++i)a[i]=i; for(i=1;i<=n;++i) solve(i); for(i=1;i<=m;++i)printf("%d\n",ans[i]); return 0; }