PS
232. Implement Queue using Stacks (leetcode)
깊게 생각하고 최선을 다하자
2022. 6. 21. 21:06
https://leetcode.com/problems/implement-queue-using-stacks/
1. 문제에 대한 이해
- 우리가 풀어야 할 문제는 무엇인가?
- 주어진 자료는 무엇인가?
- 조건은 무엇인가?
- 우리가 문제를 풀기 위해 주어진 자료가 충분한가?
- 숨겨진 조건이나 자료가 있는가? 그렇다면 그 것을 다른 방법으로 해석해보라.
- 우리가 풀어야 할 문제는 무엇인가?
-> 두 개의 스택을 사용해서 큐 구현하기
- 주어진 자료는 무엇인가?
-> push() 메소드, pop() 메소드, peek() 메소드, empty() 메소드
- 조건은 무엇인가?
->
2. 계획
- 전에 비슷한 문제를 알고 있는가?
- 이 문제를 푸는데 있어서 유용하게 쓸 수 있는 지식은 무엇인가?
- 비슷한 문제를 풀어본 적이 있다면 그 것을 활용할 수 있는가?
- 만약 문제를 풀 수 없다면 문제를 더 단순하게 하기 위해서 주어진 조건을 버려보아라
- 주어진 자료로부터 유용한 것을 이끌어 낼 수 있는가?
- 자료는 모두 사용했는가?
- 조건을 모두 사용했는가?
- 문제에 포함된 핵심적인 개념은 모두 고려했는가?
- 전에 비슷한 문제를 알고 있는가?
-> stack 2개를 사용한다
- 이 문제를 푸는데 있어서 유용하게 쓸 수 있는 지식은 무엇인가?
-> 스택을 2(stk1, stk2)개 선언하고,
push는 stk1에 하고,
peek이나 pop을 하기 전에 stk1에 있는 것을 stk2로 옮겨서 peek, pop을 하고,
다시 stk2에서 stk1으로 옮긴다.
3. 실행
- 풀이 계획을 실행하고, 각 단계가 올바른지 점검하라.
import java.util.*;
class MyQueue {
Stack<Integer> stk1;
Stack<Integer> stk2;
public MyQueue() {
stk1 = new Stack<>();
stk2 = new Stack<>();
}
public void push(int x) {
stk1.push(x);
return;
}
public int pop() {
int result = 0;
while(!stk1.isEmpty()){
int num = stk1.pop();
stk2.push(num);
}
result = stk2.pop();
while(!stk2.isEmpty()){
int num = stk2.pop();
stk1.push(num);
}
return result;
}
public int peek() {
int result = 0;
while(!stk1.isEmpty()){
int num = stk1.pop();
stk2.push(num);
}
result = stk2.peek();
while(!stk2.isEmpty()){
int num = stk2.pop();
stk1.push(num);
}
return result;
}
public boolean empty() {
if(stk1.isEmpty()){
return true;
}else{
return false;
}
}
}
4. 반성
- 문제를 다른 방식으로 해결할 수 있는가?
- 결과나 방법을 어떤 다른 문제에 활용할 수 있는가?
- 어떻게 하면 더 효율적으로 문제를 해결할 수 있는가?
- 어떻게 하면 더 효과적으로 문제를 해결할 수 있는가?
- 문제를 다른 방식으로 해결할 수 있는가?
- 어떻게 하면 더 효율적으로 문제를 해결할 수 있는가?
- enqueue()의 시간 복잡도는 무엇인가?
-> O(1)이다.
- dequeue()의 시간 복잡도는 무엇인가?
-> amortized O(1)이다.
-> 즉, stk2가 비었을 때는 O(n), stk2가 비어 있지 않을 때는 O(1)이다.
[코드 리팩토링]
- 굳이 stk2에 있는 것을 stk1으로 다시 옮길 필요가 없다
-> 단, stk2가 비어 있을 때만, stk1의 것을 옮기면 된다.
-> 그리고 empty()에 대한 검사 코드 로직도 변경된다.
import java.util.*;
class MyQueue {
Stack<Integer> stk1;
Stack<Integer> stk2;
public MyQueue() {
stk1 = new Stack<>();
stk2 = new Stack<>();
}
public void push(int x) {
stk1.push(x);
return;
}
public int pop() {
if(stk2.isEmpty()){
while(!stk1.isEmpty()){
int num = stk1.pop();
stk2.push(num);
}
}
return stk2.pop();
}
public int peek() {
if(stk2.isEmpty()){
while(!stk1.isEmpty()){
int num = stk1.pop();
stk2.push(num);
}
}
return stk2.peek();
}
public boolean empty() {
return stk1.isEmpty() && stk2.isEmpty();
}
}