2750: [HAOI2012]Road

题目链接

题目大意：有向图，一条路径被称为最短路，当且仅当不存在从它的起点到终点的另外一条路径总长度比它小，两条最短路不同，当且仅当它们包含的道路序列不同，询问每条边经过的不同的最短路条数

题解：蒟蒻只会floyd解法……简单暴力$O(n^2m)$也过不了……
Orz了题解后发现dijkstra可以在求最短路的同时将点按距源点的距离排序……
依据这个，从前往后枚举在最短路上的边可以得到源点到每个点的最短路的数目，记为a[i]
从后往前枚举在最短路上的边可以得到经过每个点有多少条最短路，记为b[i]

对于每条边，若其在最短路上，ans+=a[e[i].from]*b[e[i].go]

我的收获：Orz

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;

#define P 1000000007
#define INF 0x3f3f3f3f
#define MP(x,y) make_pair(x,y)
const int M=1505;
typedef pair<int,int> pii;

bool vis[M];
int n,m,t,head[M],d[M],c[M];
long long a[M],b[M],ans[5005];

priority_queue<pii,vector<pii>,greater<pii> > q;

struct edge{int fro,to,val,nex,id;}e[5005];

void add(int u,int v,int w,int i){e[t]=(edge){u,v,w,head[u],i};head[u]=t++;}

inline void U(long long &x,int v){x+=v;if(x>=P) x-=P;}

void dijkstra(int s)
{
    int tot=0;
    for(int i=1;i<=n;i++) vis[i]=0,d[i]=INF;
    d[s]=0;q.push(MP(d[s],s));
    while(!q.empty()){
        int u=q.top().second;q.pop();
        if(vis[u]) continue;vis[u]=1;c[++tot]=u;
        for(int v,i=head[u];i!=-1;i=e[i].nex){
            if(d[v=e[i].to]>d[u]+e[i].val){
                d[v]=d[u]+e[i].val;
                q.push(make_pair(d[v],v));
            }
        }
    }
    for(int i=1;i<=n;i++) a[i]=0,b[i]=0;
    a[s]=1;for(int i=1;i<=tot;i++) b[c[i]]=1;

    for(int x=1;x<=tot;x++) for(int i=head[c[x]];i!=-1;i=e[i].nex) 
    if(d[c[x]]+e[i].val==d[e[i].to]) U(a[e[i].to],a[c[x]]);

    for(int x=tot;x;x--) for(int i=head[c[x]];i!=-1;i=e[i].nex) 
    if(d[c[x]]+e[i].val==d[e[i].to]) U(b[c[x]],b[e[i].to]);

    for(int i=0;i<t;i++) if(d[e[i].fro]+e[i].val==d[e[i].to])
    U(ans[e[i].id],a[e[i].fro]*b[e[i].to]);
}

void work()
{
    for(int i=1;i<=n;i++) dijkstra(i);
    for(int i=1;i<=m;i++) printf("%lld\n",ans[i]);
}

void init()
{
    int x,y,z;
    cin>>n>>m;memset(head,-1,sizeof(head));
    for(int i=1;i<=m;i++) scanf("%d%d%d",&x,&y,&z),add(x,y,z,i);
}

int main()
{
    init();
    work();
    return 0;
}