1. 程式人生 > >【劍指offer python】面試題7:用兩個棧實現佇列

【劍指offer python】面試題7:用兩個棧實現佇列

題目連結

題目描述

用兩個棧來實現一個佇列,完成佇列的Push和Pop操作。 佇列中的元素為int型別。

solution

class Solution:
    
    def __init__(self):
        self.stack1 = []
        self.stack2 = []
    def push(self, node):
        self.stack1.append(node)
        # write code here
    def pop(self):
        if self.stack2 == []:
            for i in range(len(self.stack1)):
                self.stack2.append(self.stack1.pop())
        return self.stack2.pop()