# 유형 : 다익스트라, 그래프
# 난이도 : 골드 V
# 맨 처음에는 인접리스트, 우선순위큐 대신 2차원 배열과 반복문을 사용해서 메모리 초과가 났다. 그 이유는 Vertex 최대 개수는 20000, 따라서 20000*20000*4Byte가 필요한 가중치 그래프, 20000*4Byte 의 Dist 그래프, 20000 * 1Byte boolean그래프 때문이었다. 검색해보니 다익스트라를 풀 때는 우선순위큐와 ArrayList를 사용하는게 좋다고한다.
|
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
package bj;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.PriorityQueue;
import java.util.StringTokenizer;
public class p1753 {
static int V,E,Start,INF=Integer.MAX_VALUE;
static int dist[];
static ArrayList<Point> arrList[];
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
V = Integer.parseInt(st.nextToken());
E = Integer.parseInt(st.nextToken());
arrList = new ArrayList[V+1];
dist = new int[V+1];
for(int i=1; i<=V; i++)
arrList[i] = new ArrayList<>();
for(int i=1; i<=V; i++)
dist[i] = INF;
Start = Integer.parseInt(br.readLine());
dist[Start] = 0;
for(int e=0; e<E; e++) {
st = new StringTokenizer(br.readLine());
int u = Integer.parseInt(st.nextToken());
int v = Integer.parseInt(st.nextToken());
int val = Integer.parseInt(st.nextToken());
arrList[u].add(new Point(v, val));
}
solve();
for(int i=1; i<=V; i++) {
if(dist[i] == INF)
System.out.println("INF");
else
System.out.println(dist[i]);
}
}
public static void solve() {
PriorityQueue<Point> pq = new PriorityQueue<>();
// PriorityQueue = (Vertex, weight)
pq.add(new Point(Start, 0));
while(!pq.isEmpty()) {
Point po = pq.poll();
// 갱신하려고 하는 값이 PQ에서 poll한 노드의 값보다 작을 땐 갱신 X
if(dist[po.vertex] < po.val)
continue;
// 현재 노드에 연결된 Vertex가 저장 된 ArrayList를 반복문을 통해 탐색하며 더 작은 값으로 갱신할 수 있다면 갱신하면서 PQ에 넣어준다.
for(int i=0; i<arrList[po.vertex].size(); i++) {
int tmpIndex = arrList[po.vertex].get(i).vertex;
int tmpDist = po.val + arrList[po.vertex].get(i).val;
if(dist[tmpIndex] > tmpDist) {
dist[tmpIndex] = tmpDist;
pq.add(new Point(tmpIndex, tmpDist));
}
}
}
}
// 우선순위큐를 편하게 사용하기 위해 정렬기준 변경
public static class Point implements Comparable<Point>{
int vertex;
int val;
public Point(int v, int val) {
this.vertex=v;
this.val=val;
}
@Override
public int compareTo(Point o) {
if(this.val > o.val)
return 1;
// TODO Auto-generated method stub
return 0;
}
}
}
|
cs |
'백준' 카테고리의 다른 글
| #백준_1058 친구 - Java 자바 (0) | 2020.04.17 |
|---|---|
| #백준_2665 미로만들기 - Java 자바 (0) | 2020.04.14 |
| #백준_10835 카드게임 - Java 자바 (0) | 2020.04.04 |
| #백준_17142 연구소 3 - Java 자바 (0) | 2020.04.03 |
| #백준_16954 움직이는 미로 탈출 - Java 자바 (0) | 2020.04.01 |
