刷题记录（NC19987 [HAOI2012]ROAD）

NC19987 [HAOI2012]ROAD

题目链接

关键点：

1、题目大意：求出所有边上经过的最短路（起点终点自定）数量

2、求出所有的最短路，用dijkstra算法求出以所有点为起点的最短路。

3、对于所经过的最短路数量，假如一条边为最短路所经过的，那么起点到该边起点的最短路的数量num，和该边终点到终点的最短路数量num2，两个数量相乘即为该边最短路的数量。通俗的讲，一点的num为起点到该点的最短路数量，num2为该点到终点的最短路的数量

4、在用dijkstra求最短路的过程中，一旦确定一边为最短路上的组成部分，那么就更新num数量；求num2,可以利用dfs来求，从起点出发，起点的num2即为从起点到终点最短路上的所有边

5、总结：对所有点都跑一遍dijkstra和dfs，每边的最短路数量ans[i]+=num[该边的起点]*num2[该点的终点](每次以不同的点作为起点时，每次更新)

完整代码：

# include <bits/stdc++.h>
using namespace std;
struct ty{
    int t, next, len;
}edge[10000+10];
struct ty2{
    int x, len;
    bool operator < (struct ty2 a) const{
        return len>a.len;
    }
};
const int MOD = 1000000007;
int n, m, cnt;
int head[5010], dist[1500+10], vis[1500+10], num[1500+10], num2[1500+10];
long long ans[5000+10];
priority_queue<ty2> q;
void addedge(int x, int y, int len)
{
    edge[++cnt].t = y;
    edge[cnt].len = len;
    edge[cnt].next = head[x];
    head[x] = cnt;
}
void dis(int s)
{
    ty2 tmp;
    tmp.len = 0, tmp.x = s;
    q.push(tmp);
    int x = tmp.x;
    dist[x] = 0;
    num[x] = 1;
    while (!q.empty())
    {
        ty2 tmp = q.top();
        q.pop();
        int x = tmp.x;
        if (vis[x]) continue;
        vis[x] = 1;
        for (int i = head[x]; i!=-1; i = edge[i].next)
        {
            int y = edge[i].t;
            if (vis[y]) continue;
            if (dist[x]+edge[i].len == dist[y])
            {
                num[y] = (num[y]+num[x])%MOD;
            }
            else if (dist[x]+edge[i].len < dist[y])
            {
                dist[y] = dist[x]+edge[i].len;
                ty2 tmp;
                tmp.x = y, tmp.len = dist[y];
                num[y] = num[x];
                q.push(tmp);
            }
        }
    }
}
void dfs(int s)
{
    if (num2[s]) return ;
    num2[s] = 1;
    for (int i=head[s]; i!=-1; i=edge[i].next)
    {
        int y = edge[i].t;
        if (dist[s] + edge[i].len == dist[y])
        {
            dfs(y);
            num2[s] += num2[y];
            ans[i] = (ans[i] + (1ll*num[s]*num2[y])%MOD)%MOD;
        }
    }
}
int main()
{
    memset(head, -1, sizeof(head));
    cin>>n>>m;
    for (int i=1; i<=m; i++)
    {
        int x, y, len;
        cin>>x>>y>>len;
        addedge(x, y, len);
    }
    for (int i=1; i<=n; i++)
    {
        memset(vis, 0, sizeof(vis));
        memset(num, 0, sizeof(num));
        memset(num2, 0, sizeof(num2));
        memset(dist, 0x3f, sizeof(dist));
        dis(i);
        dfs(i);
    }
    for (int i=1; i<=m; i++)
        cout<<ans[i]<<endl;
    
    
    return 0;
}