Road（bzoj 2750）

/*
        刚开始打的暴力，由于细节问题，挂成了10分，后来能改到50分。
        一种很容易想到的方法是DP，但是很无奈的是想不到一种更新方法能够保证拓扑序，然而Dijkstral可以记录图的拓扑序。 
        记录每个节点前后各经过了几次，然后用乘法原理得到答案。 
*/
#include<cstdio>
#include<iostream>
#include<queue>
#include<algorithm>
#include<cstring>
#define N 1510
#define M 5010
#define mod 1000000007
#define inf 1000000000
using namespace std;
int head[N],dis[N],inq[N],a[N],b[N],c[N],sum[M],n,m;
struct node{int v,w,pre;}e[M];
struct Node{
    int id,dist;
    bool operator< (const Node&s1) const {
        return s1.dist<dist;
    }
};priority_queue<Node> q;

void add(int i,int u,int v,int w){
    e[i].v=v;e[i].w=w;e[i].pre=head[u];head[u]=i;
}
void Dij(int S){
    memset(a,0,sizeof(a));
    memset(b,0,sizeof(b));
    memset(inq,0,sizeof(inq));
    for(int i=1;i<=n;i++) dis[i]=inf;int cnt=0;
    dis[S]=0;q.push((Node){S,0});
    while(!q.empty()){
        int x=q.top().id;q.pop();
        if(inq[x]) continue;
        inq[x]=1;c[++cnt]=x;
        for(int i=head[x];i;i=e[i].pre)
            if(dis[e[i].v]>dis[x]+e[i].w){
                dis[e[i].v]=dis[x]+e[i].w;
                q.push((Node){e[i].v,dis[e[i].v]});
            }
    }
    for(int i=1;i<=cnt;i++) b[c[i]]=1;
    a[S]=1;
    for(int i=1;i<=cnt;i++)
        for(int j=head[c[i]];j;j=e[j].pre)
            if(dis[c[i]]+e[j].w==dis[e[j].v]) a[e[j].v]+=a[c[i]],a[e[j].v]%=mod;
    for(int i=cnt;i;i--)
        for(int j=head[c[i]];j;j=e[j].pre)
            if(dis[c[i]]+e[j].w==dis[e[j].v]) b[c[i]]+=b[e[j].v],b[c[i]]%=mod;
    for(int i=1;i<=n;i++)
        for(int j=head[i];j;j=e[j].pre)
            if(dis[i]+e[j].w==dis[e[j].v]) sum[j]=(sum[j]+(long long)a[i]*b[e[j].v])%mod;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        int u,v,w;scanf("%d%d%d",&u,&v,&w);
        add(i,u,v,w);
    }
    for(int i=1;i<=n;i++) Dij(i);
    for(int i=1;i<=m;i++) printf("%d\n",sum[i]);
    return 0;
}