forked from onlybooks/java-algorithm-interview
-
Notifications
You must be signed in to change notification settings - Fork 0
/
P24_2.kt
40 lines (34 loc) · 1.13 KB
/
P24_2.kt
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
package ch09
import java.util.*
class P24_2 {
// 삽입할 때 사용하는 스택 선언
val input: Deque<Int> = ArrayDeque()
// 추출할 때 사용하는 스택 선언
val output = ArrayDeque<Int>()
fun push(x: Int) {
// 삽입은 삽입 스택에 단순 추가
input.push(x)
}
fun pop(): Int {
// 추출 스택 조회하면서 비어 있다면 처리 진행
peek()
// 추츨 스택에 있는 마지막 값 추출
return output.pop()
}
fun peek(): Int {
// 추출 스택에 저장된 게 없다면 진행
if (output.isEmpty()) {
// 삽입 스택이 비워질 때까지 진행
while (!input.isEmpty()) {
// 삽입 스택에서 추출한 결과를 추출 스택에 삽입(역순으로 저장된다.)
output.push(input.pop())
}
}
// 가장 나중에 삽입된 결과 조회
return output.peek()
}
fun empty(): Boolean {
// 두 스택이 모두 비어야 비어 있는 것으로 처리
return input.isEmpty() && output.isEmpty()
}
}