1976. 到达目的地的方案数Java
文章发布较早,内容可能过时,阅读注意甄别。
# 题目
你在一个城市里,城市由 n 个路口组成,路口编号为 0 到 n - 1 ,某些路口之间有 双向 道路。输入保证你可以从任意路口出发到达其他任意路口,且任意两个路口之间最多有一条路。
给你一个整数 n 和二维整数数组 roads ,其中 roads[i] = [ui, vi, timei] 表示在路口 ui 和 vi 之间有一条需要花费 timei 时间才能通过的道路。你想知道花费 最少时间 从路口 0 出发到达路口 n - 1 的方案数。
请返回花费 最少时间 到达目的地的 路径数目 。由于答案可能很大,将结果对 109 + 7 取余 后返回。
示例 1:
输入:n = 7, roads = [[0,6,7],[0,1,2],[1,2,3],[1,3,3],[6,3,3],[3,5,1],[6,5,1],[2,5,1],[0,4,5],[4,6,2]]
输出:4
解释:从路口 0 出发到路口 6 花费的最少时间是 7 分钟。
四条花费 7 分钟的路径分别为:
- 0 ➝ 6
- 0 ➝ 4 ➝ 6
- 0 ➝ 1 ➝ 2 ➝ 5 ➝ 6
- 0 ➝ 1 ➝ 3 ➝ 5 ➝ 6
示例 2:
输入:n = 2, roads = [[1,0,10]]
输出:1
解释:只有一条从路口 0 到路口 1 的路,花费 10 分钟。
提示:
- 1 <= n <= 200
- n - 1 <= roads.length <= n * (n - 1) / 2
- roads[i].length == 3
- 0 <= ui, vi <= n - 1
- 1 <= timei <= 109
- ui != vi
- 任意两个路口之间至多有一条路。
- 从任意路口出发,你能够到达其他任意路口。
# 思路
LinkedList
# 解法
class Solution {
public int countPaths(int n, int[][] roads) {
int mod = 1000000000 + 7, inf = Integer.MAX_VALUE;
Map<Integer, List<int[]>> map = new HashMap<>();
for (int i = 0; i < n; i++) {
map.put(i, new ArrayList<>());
}
for (int[] edge : roads) {
int u = edge[0],
v = edge[1],
w = edge[2];
map.get(u).add(new int[]{v, w});
map.get(v).add(new int[]{u, w});
}
int[] dis = new int[n];
boolean[] use = new boolean[n];
int[] pre = new int[n];
Arrays.fill(pre, 1);
LinkedList<int[]> query = new LinkedList<>();
Arrays.fill(dis, inf);
dis[0] = 0;
query.add(new int[]{0, 0});
while (!query.isEmpty()) {
int[] u = query.poll();
if (use[u[0]]) continue;
use[u[0]] = true;
List<int[]> edges = map.get(u[0]);
if (edges != null) {
for (int[] edge : edges) {
int to = edge[0];
if (use[to]) {
continue;
} else if (dis[to] > u[1] + edge[1]) {
pre[to] = pre[u[0]];
dis[to] = u[1] + edge[1];
query.offer(new int[]{to, dis[to]});
} else if (dis[to] == u[1] + edge[1]) {
pre[to] += pre[u[0]];
pre[to] %= mod;
}
}
query.sort(new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
return o1[1] - o2[1];
}
});
}
}
// System.out.println(Arrays.toString(dis));
return pre[n - 1];
}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
# 总结
- 分析出几种情况,然后分别对各个情况实现