Python/알고리즘

[파이썬 알고리즘 인터뷰] 23. 큐를 이용한 스택 구현

AI 꿈나무 2021. 3. 2. 15:30
반응형

 

 이 포스팅은 파이썬 알고리즘 인터뷰를 공부하면서 정리한 내용입니다.

 

 출처 : 파이썬 알고리즘 인터뷰

 코드 출처 : 파이썬 알고리즘 인터뷰 깃허브

 


23. 큐를 이용한 스택 구현

leetcode 225. Implement Stack using Queues 문제입니다.

leetcode.com/problems/implement-stack-using-queues/

 

 

풀이

class MyStack:
    def __init__(self):
        self.q = collections.deque()

    def push(self, x):
        self.q.append(x)
        # 요소 삽입 후 맨 앞에 두는 상태로 재정렬
        for _ in range(len(self.q) - 1):
            self.q.append(self.q.popleft())

    def pop(self):
        return self.q.popleft()

    def top(self):
        return self.q[0]

    def empty(self):
        return len(self.q) == 0
반응형