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();
    }
}