9주차. 스택,큐(스택을 이용한 큐 구현)

24. 스택을 이용한 큐 구현

https://leetcode.com/problems/implement-queue-using-stacks/

Implement Queue using Stacks - LeetCode

📌문제
스택을 이용해다음연산을 지원하는 큐를 구현하라

• push(x): 요소 x를 큐 마지막에 삽입한다.
• pop(): 큐 처음에 있는 요소를 제거한다.
• peek(): 큐 처음에 있는 요소를 조회한다.
• empty(): 큐가 비어 있는지 여부를 리턴한다.  

- 예제1
  📝입력

["MyQueue", "push", "push", "peek", "pop", "empty"]
[[], [1], [2], [], [], []]

  💻출력

[null, null, null, 1, 1, false]

 

    
📌풀이

0ms, 40.3mb

class MyQueue {

    Stack<Integer> input = new Stack();
    Stack<Integer> output = new Stack();

    public MyQueue() {
        
    }
    
    public void push(int x) {
        input.push(x);
    }

    public int pop() {
        peek();
        return output.pop();
    }

    public int peek() {
        if (output.empty())
            while (!input.empty())
                output.push(input.pop());
        return output.peek();
    }

    public boolean empty() {
        return input.empty() && output.empty();
    }
}

스택 두 개를 사용한다는 아이디어가 떠올랐다면 쉽게 풀었을 문제.