3.4 스택으로 큐

문제 

스택 두개로 큐 하나를 구현한 MyQueue 클래스를 구현하라

 

접근법

q1, q2가 존재하고 push를 하면 q1에 넣는다.

pop를 부르는 순간, p1에서 p2에 순차적으로 pop하고 push한다. 

그리고 p2에서 pop하면 가장 먼저 들어간 데이터가 먼저 나올 것이다. (FIFO)

class QueueByStack :
    def __init__(self):
        s1 = []
        s2 = []

    def push(self, num):
        sl.append(num)

    def pop(self):
        if not s2 :
            while s1 :
                s2.append(s1.pop())
        return s2.pop()