bzoj2750: [HAOI2012]Road

2750: [HAOI2012]Road

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 651  Solved: 302
[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

 

 

建出最短路图。。然后答案就是起点到这个点的最短路径乘以这个点到终点的最短路径。。

但是觉得挺难写的TAT

然后发现了dij的新姿势。。

扔个链接 http://www.cnblogs.com/zyfzyf/p/3995257.html

dijkstra算法可以在算出最短路的同时将点的源点的距离排序，然后按照这个

从前往后枚举在最短路上的边可以得到源点到每个点的最短路的数目，记为a[i]

从后往前枚举在最短路上的边可以得到经过每个点有多少条最短路，记为b[i] 

然后对于每条边就是 +=a[e[i].from]*b[e[i].go]

#include<bits/stdc++.h>
#define rep(i,l,r) for(int i=l;i<=r;++i)
#define pa pair<int,int>
using namespace std;
const int N=5005;
typedef long long ll;
const ll ccz=1000000007;
int head[N],tot,dis[N],u,v,w,n,m,cnt,c[N];
ll a[N],b[N],ans[N];
struct zs{
    int to,next,w,id;
}e[N];
bool in[N];
inline void ins(int u,int v,int w,int id){
    e[++tot].to=v; e[tot].next=head[u]; head[u]=tot; e[tot].w=w; e[tot].id=id;
}
inline void run(int s){
    priority_queue<pa,vector<pa>,greater<pa> >q;
    memset(in,0,sizeof in);
    memset(dis,60,sizeof dis); dis[s]=0; q.push(make_pair(0,s));
    cnt=0;
    while(!q.empty()){
        int x=q.top().second; q.pop();
        if(in[x]) continue; in[x]=1; c[++cnt]=x;
        for(int k=head[x];k;k=e[k].next) if(dis[x]+e[k].w<dis[e[k].to]) {
            dis[e[k].to]=e[k].w+dis[x];
            q.push(make_pair(dis[e[k].to],e[k].to));
        }
    }
    memset(a,0,sizeof a); memset(b,0,sizeof b);
    rep(i,1,cnt) b[c[i]]=1;
    a[s]=1;
    rep(i,1,cnt) for(int k=head[c[i]];k;k=e[k].next) if(dis[c[i]]+e[k].w==dis[e[k].to])(a[e[k].to]+=a[c[i]])%=ccz;
    for(int i=cnt;i;i--) for(int k=head[c[i]];k;k=e[k].next) if(dis[c[i]]+e[k].w==dis[e[k].to])(b[c[i]]+=b[e[k].to])%=ccz;
    rep(i,1,n) for(int k=head[i];k;k=e[k].next) if(dis[i]+e[k].w==dis[e[k].to])(ans[e[k].id]+=a[i]*b[e[k].to])%=ccz;
}
int main(){
    scanf("%d%d",&n,&m);
    rep(i,1,m) scanf("%d%d%d",&u,&v,&w),ins(u,v,w,i);
    rep(i,1,n) run(i);
    rep(i,1,m) printf("%lld\n",ans[i]);
}