offer通過--7兩個棧實現佇列-2

獨鹿發表於2018-05-02
# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack_1 = []
        self.stack_2 = []
    def push(self, node):
        # write code here
        self.stack_1.append(node)
    def pop(self):
        # return xx
        if self.stack_2:
            return self.stack_2.pop()
        while self.stack_1:
            self.stack_2.append(self.stack_1.pop())
        if self.stack_2:
            return self.stack_2.pop()

相關文章