BZOJ 2750: [HAOI2012]Road
阿新 • • 發佈:2018-11-15
其實也就是個最短路。。。
以每個點為起點 都跑一次最短路 然後統計一下就好了。。
對於在最短路圖上的邊(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;
}