forked from ohxxx/algorithm-notes
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdijkstra.js
More file actions
60 lines (50 loc) · 1.33 KB
/
Copy pathdijkstra.js
File metadata and controls
60 lines (50 loc) · 1.33 KB
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
56
57
58
59
60
const INF = Number.MAX_SAFE_INTEGER
const minDistance = (dist, visited) => {
let min = INF
let minIndex = -1
for (let v = 0; v < dist.length; v++) {
if (visited[v] === false && dist[v] <= min) {
min = dist[v]
minIndex = v
}
}
return minIndex
}
const dijkstra = (graph, src) => {
const dist = []
const visited = []
const { length } = graph
for (let i = 0; i < length; i++) { // 把所有距离(dist)、标记点(visited) 初始化
dist[i] = INF
visited[i] = false
}
dist[src] = 0 // 把源顶点到自己的距离设为0
for (let i = 0; i < length - 1; i++) { // 找出到其余顶点的最短路径
const u = minDistance(dist, visited) // 需要从尚未处理的顶点中选出距离最近的顶点
visited[u] = true // 把选出的顶点标为visited
for (let v = 0; v < length; v++) {
if (
!visited[v] &&
graph[u][v] !== 0 &&
dist[u] !== INF &&
dist[u] + graph[u][v] < dist[v]
) { // 找到更短的路径
dist[v] = dist[u] + graph[u][v]
}
}
}
return dist
}
// test
const graph = [
[0, 2, 4, 0, 0, 0],
[0, 0, 2, 4, 2, 0],
[0, 0, 0, 0, 3, 0],
[0, 0, 0, 0, 0, 2],
[0, 0, 0, 3, 0, 2],
[0, 0, 0, 0, 0, 0]
]
const dist = dijkstra(graph, 0)
for (i = 0; i < dist.length; i++) {
console.log(i + '\t\t' + dist[i])
}