P2505 [HAOI2012]道路

题目描述

C国有n座城市,城市之间通过m条[b]单向[/b]道路连接。一条路径被称为最短路,当且仅当不存在从它的起点到终点的另外一条路径总长度比它小。两条最短路不同,当且仅当它们包含的道路序列不同。我们需要对每条道路的重要性进行评估,评估方式为计算有多少条不同的最短路经过该道路。现在,这个任务交给了你。

输入输出格式

输入格式:

第一行包含两个正整数n、m

接下来m行每行包含三个正整数u、v、w,表示有一条从u到v长度为w的道路

输出格式:

输出应有m行,第i行包含一个数,代表经过第i条道路的最短路的数目对[b]1000000007取模[/b]后的结果

输入输出样例

输入样例#1: 复制
4 4
1 2 5
2 3 5
3 4 5
1 4 8
输出样例#1: 复制
2
3
2
1

说明

数据规模

30%的数据满足:n≤15、m≤30

60%的数据满足:n≤300、m≤1000

100%的数据满足:n≤1500、m≤5000、w≤10000





 

开始想了一波枚举每一条边,对于。。。。。。。。。  

然后就没有想出来,看题解吧qwq

看完题解后我是震惊的

每一个点都spfa跑最短路,然后就会有最短路树,在最短路树上,对于每一条边,加上源点到这条边起点的最短路径条数×经过终点的最短路径条数,因为无环,topsort,按拓扑序第一个正退,第二个逆推

看数据范围 点数×o(spfa) 这都能过?? 而且最慢的点400ms,这。。。,根本就不敢这么写好吗 = =|;

然后,带权最短路计数不能用spfa,QAQ;





 

 1 #include <cmath>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <iostream>
 5 #include <algorithm>
 6 using namespace std;
 7 inline int read() {
 8     int res = 0; bool bo = 0; char c;
 9     while (((c = getchar()) < '0' || c > '9') && c != '-');
10     if (c == '-') bo = 1; else res = c - 48;
11     while ((c = getchar()) >= '0' && c <= '9')
12         res = (res << 3) + (res << 1) + (c - 48);
13     return bo ? ~res + 1 : res;
14 }
15 const int N = 1505, M = 5005, INF = 0x3f3f3f3f, PYZ = 1e9 + 7;
16 int n, m, ecnt, nxt[M], adj[N], st[M], go[M], val[M], dis[M], len, que[M << 1],
17 cnt[N], cnt1[N], cnt2[N], H, T, tot, q[N], ans[M];
18 bool vis[N], ins[M];
19 void add_edge(int u, int v, int w) {
20     nxt[++ecnt] = adj[u]; adj[u] = ecnt; st[ecnt] = u; go[ecnt] = v; val[ecnt] = w;
21 }
22 void spfa(int S) {
23     int i; memset(dis, INF, sizeof(dis));
24     memset(ins, 0, sizeof(ins));
25     dis[que[len = 1] = S] = 0;
26     for (i = 1; i <= len; i++) {
27         int u = que[i]; vis[u] = 0;
28         for (int e = adj[u], v; e; e = nxt[e])
29             if (dis[u] + val[e] < dis[v = go[e]]) {
30                 dis[v] = dis[u] + val[e];
31                 if (!vis[v]) vis[que[++len] = v] = 1;
32             }
33     }
34     for (i = 1; i <= m; i++)
35         if (dis[st[i]] + val[i] == dis[go[i]])
36             ins[i] = 1;
37 }
38 void topo(int S) {
39     memset(cnt, 0, sizeof(cnt));
40     memset(cnt1, 0, sizeof(cnt1));
41     memset(cnt2, 0, sizeof(cnt2));
42     int i; H = tot = 0; cnt1[que[T = 1] = S] = 1;
43     for (i = 1; i <= m; i++) if (ins[i]) cnt[go[i]]++;
44     while (H < T) {
45         int u = que[++H]; q[++tot] = u;
46         for (int e = adj[u], v; e; e = nxt[e]) {
47             if (!ins[e]) continue;
48             v = go[e]; if (!(--cnt[v])) que[++T] = v;
49             (cnt1[v] += cnt1[u]) %= PYZ;
50         }
51     }
52     for (i = tot; i; i--) {
53         int u = q[i]; cnt2[u]++;
54         for (int e = adj[u], v; e; e = nxt[e]) {
55             if (!ins[e]) continue;
56             (cnt2[u] += cnt2[v = go[e]]) %= PYZ;
57         }
58     }
59 }
60 void solve(int S) {
61     int i; spfa(S); topo(S);
62     for (i = 1; i <= m; i++) if (ins[i])
63         (ans[i] += 1ll * cnt1[st[i]] * cnt2[go[i]] % PYZ) %= PYZ;
64 }
65 int main() {
66     int i, x, y, z; n = read(); m = read();
67     for (i = 1; i <= m; i++) x = read(), y = read(),
68         z = read(), add_edge(x, y, z);
69     for (i = 1; i <= n; i++) solve(i);
70     for (i = 1; i <= m; i++) printf("%d\n", ans[i]);
71     return 0;
72 }

 

 

 

 

 

 

 

 

 

 

 

 

 

上一篇:HAOI2012 容易题


下一篇:最短路径:初涉Dijkstra算法