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;
}