-
Notifications
You must be signed in to change notification settings - Fork 24
/
P46_1.java
62 lines (52 loc) · 2.16 KB
/
P46_1.java
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
61
62
package ch13;
import java.util.*;
public class P46_1 {
static class Position {
final int y;
final int x;
final int distance;
public Position(int y, int x, int distance) {
this.y = y;
this.x = x;
this.distance = distance;
}
}
// 다익스트라 알고리즘을 위한 거리순 우선순위 큐 선언
Queue<Position> pq = new PriorityQueue<>(Comparator.comparingInt(o -> o.distance));
public void findPath(int y, int x, int distance, int[][] maps) {
if (y >= 0 &&
y < maps.length &&
x >= 0 &&
x < maps[0].length &&
maps[y][x] != 0) {
// 이미 지나온 경로는 더 이상 거치지 않도록 0으로 설정
maps[y][x] = 0;
// 현재 위치와 거리+1을 우선순위 큐에 삽입
pq.add(new Position(y, x, distance + 1));
}
}
public int solution(int[][] maps) {
// 출발지부터 삽입
pq.add(new Position(0, 0, 1));
// 해당 지점까지의 거리 결과를 저장하는 변수 선언
Map<Integer, Position> dist = new LinkedHashMap<>();
while (!pq.isEmpty()) {
// 거리가 가장 짧은 값 추출
Position cur = pq.poll();
// 아직 계산하지 않은 지점이라면 계산 결과 삽입
// 키/값 형태이므로 키는 (y*1000+x)로 구성
if (!dist.containsKey(cur.y * 1000 + cur.x)) {
dist.put(cur.y * 1000 + cur.x, cur);
findPath(cur.y, cur.x + 1, cur.distance, maps); // 동
findPath(cur.y, cur.x - 1, cur.distance, maps); // 서
findPath(cur.y + 1, cur.x, cur.distance, maps); // 남
findPath(cur.y - 1, cur.x, cur.distance, maps); // 북
}
}
// 마지막 위치의 결과가 있다면 해당 위치의 거리를 정답으로 리턴
if (dist.containsKey(((maps.length - 1) * 1000) + (maps[0].length - 1))) {
return dist.get(((maps.length - 1) * 1000) + (maps[0].length - 1)).distance;
}
return -1;
}
}