[bzoj2750][HAOI2012]Road

2750: [HAOI2012]Road

Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 397 Solved: 184
[Submit][Status][Discuss]
Description

C国有n座城市，城市之间通过m条单向道路连接。一条路径被称为最短路，当且仅当不存在从它的起点到终点的另外一条路径总长度比它小。两条最短路不同，当且仅当它们包含的道路序列不同。我们需要对每条道路的重要性进行评估，评估方式为计算有多少条不同的最短路经过该道路。现在，这个任务交给了你。

Input

第一行包含两个正整数n、m
接下来m行每行包含三个正整数u、v、w，表示有一条从u到v长度为w的道路

Output

输出应有m行，第i行包含一个数，代表经过第i条道路的最短路的数目对1000000007取模后的结果

Sample Input

4 4

1 2 5

2 3 5

3 4 5

1 4 8

Sample Output

2

3

2

1

HINT

数据规模

30%的数据满足：n≤15、m≤30

60%的数据满足：n≤300、m≤1000

100%的数据满足：n≤1500、m≤5000、w≤10000

题解：本来这种题可以直接floyd再加一句话就行了，可是这道题把n^3卡了，所以就得换一种方法了。
很容易可以想到我们可以枚举每个边做起点然后跑最短路，再枚举每一条变，把这条边再以i为起点的最短路中的贡献加入到答案中去。
那么这每次的贡献怎样去计算呢？
那么首先如果这条边在最短路上，那么i的贡献=当前起点到这条边的起点的最短路数*当前起点到所有点的最短路中经过当前变的终点的数目。
假设怎样求刚才说的那两个东西呢?
我们用st[i]和en[i]分别表示上面的那两个东西。
因为是有向图，所以对于st[i]我们可以用当前起点往后递推出每一个点的值。在递推的过程中我们需要保证当前这个点之前的点的情况已经全部处理完了。
那么就需要再处理出一个pre[i]表示i有几个最短路上面的前驱，这样当pre[i]==0的时候我们就可以向下扩展了。
然后再说en[i]由于这个东西是相当于处理的后面的信息，所以我们可以先处理出它后面的所有节点的信息，再来更新这个节点，相当于倒着更新。这个判断后面节点是否全部更新过了就比较简单了，随便判断一下就行了。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define D 1000000007
#define LL long long
const int N=1600;
const int M=5100;
bool f[N],check[N];
LL ans[M],st[M],en[M];
struct S{int st,en,va;}aa[M],e[M];
int n,m,tot,point[N],next[M],dis[N],l[M*10],pre[N];
inline void add(int x,int y,int z){
    tot+=1;next[tot]=point[x];point[x]=tot;
    aa[tot].st=x;aa[tot].en=y;aa[tot].va=z;
}
inline void SPFA(int x){
    int i,h,t,u;
    memset(f,1,sizeof(f));
    memset(dis,127/3,sizeof(dis));
    h=t=1;l[h]=x;dis[x]=0;
    while(h<=t){
        u=l[h]; f[u]=true;
        for(i=point[u];i;i=next[i])
          if(dis[aa[i].en]>dis[u]+aa[i].va){
            dis[aa[i].en]=dis[u]+aa[i].va;
            if(f[aa[i].en]){
                f[aa[i].en]=false;
                l[++t]=aa[i].en;
            }
          }
        h+=1;
    }
}
inline void get_pre(int x){
    int i;
    f[x]=false;
    for(i=point[x];i;i=next[i])
      if(dis[aa[i].en]==dis[x]+aa[i].va){
        pre[aa[i].en]+=1;
        if(f[aa[i].en]) get_pre(aa[i].en);
      }
}
inline void get_st(int x){
    int i;
    for(i=point[x];i;i=next[i])
      if(dis[aa[i].en]==dis[x]+aa[i].va){
        st[aa[i].en]=(st[aa[i].en]+st[x])%D;
        pre[aa[i].en]-=1;
        if(!pre[aa[i].en]) get_st(aa[i].en);
      }
}
inline void get_en(int x){
    int i;
    en[x]=1;
    for(i=point[x];i;i=next[i])
      if(dis[aa[i].en]==dis[x]+aa[i].va){
        if(!en[aa[i].en]) get_en(aa[i].en);
        en[x]=(en[aa[i].en]+en[x])%D;
      }
}
int main(){
    int i,j,x,y,z;
    scanf("%d%d",&n,&m);
    for(i=1;i<=m;++i){
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
        e[i].st=x;e[i].en=y;e[i].va=z;
    }
    for(i=1;i<=n;++i){
        SPFA(i);
        memset(f,1,sizeof(f));
        memset(st,0,sizeof(st));
        memset(en,0,sizeof(en));
        get_pre(i);
        st[i]=1; get_st(i);
        get_en(i);
        for(j=1;j<=m;j++)
          if(dis[e[j].st]+e[j].va==dis[e[j].en]) 
            ans[j]=(ans[j]+st[e[j].st]*en[e[j].en])%D;
    }
    for(i=1;i<=m;++i) 
      printf("%lld\n",ans[i]);
}