bzoj2750 [HAOI2012]Road

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

一道最短路的变形题
如果一条边i在最短路上，那么i的贡献=当前起点到这条边的起点的最短路数*当前起点到所有点的最短路中经过当前变的终点的数目。
那么设当前起点到这条边的起点的最短路数为st[ ]，当前起点到所有点的最短路中经过当前变的终点的数目为ed[ ]，然后每个点在最短路上的前驱为pre[ ]，对于每个点跑一边spfa，然后每次求出每个点的st，ed，pre，ans+=st[u]*ed[v]。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define N 1510
#define M 5010
#define mod 1000000007
using namespace std;
int n,m,head[N],dis[N],pre[N],st[M],ed[M],ans[M]; 
bool used[N];

struct Node{
    int to,nxt,w,frm;
}edge[M];

void add(int j,int x,int y,int z)
{
    edge[j].nxt=head[x];
    edge[j].to=y;
    edge[j].frm=x;
    edge[j].w=z;
    head[x]=j;
}

 void SPFA(int s)
 {
    memset(dis,0x3f,sizeof dis); memset(used,0,sizeof used);
    queue<int> q;
    dis[s]=0;
    q.push(s);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        used[u]=0;
        for(int i=head[u];i;i=edge[i].nxt)
        {
            int v=edge[i].to;
            if(dis[v]>dis[u]+edge[i].w)
            {
                dis[v]=dis[u]+edge[i].w;
                if(!used[v])
                {
                    used[v]=1;
                    q.push(v);
                }
            }
         }
     }
 }

 void get_pre(int x)
 {
    used[x]=1;
    for(int i=head[x];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(dis[v]==dis[x]+edge[i].w)
        {
            pre[v]++;
            if(!used[v]) get_pre(v);
         }
     }
 }

 void get_st(int x)
 {
    for(int i=head[x];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(dis[v]==dis[x]+edge[i].w)
        {
            st[v]+=st[x];
            st[v]%=mod;
            pre[v]--;
            if(!pre[v]) get_st(v);
         }
     }
}

void get_ed(int x)
{
    ed[x]=1;
    for(int i=head[x];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(dis[v]==dis[x]+edge[i].w)
        {
            if(!ed[v]) get_ed(v);
            ed[x]+=ed[v];
            ed[x]%=mod;
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(i,x,y,z);
    }
    for(int i=1;i<=n;i++)
    {
        SPFA(i);
        memset(used,0,sizeof used);
        memset(st,0,sizeof st);
        memset(ed,0,sizeof ed);
        get_pre(i);
        st[i]=1; get_st(i);
        get_ed(i);
        for(int j=1;j<=m;j++)
        {
            int u=edge[j].frm,v=edge[j].to;
            if(dis[v]==dis[u]+edge[j].w)
            {
                ans[j]+=st[u]*ed[v];
                ans[j]%=mod;
            }
        }
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
}