剑指offer 用两个栈实现队列 @python

题目描述

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

Solution

• 用一个栈实现进队，另一个栈实现出队
• 需要进队的时候把元素压如stack1中，需要出队的时候把stack1中的元素全部弹出至stack2中
• 从stack2中出队，即可实现先进先出

# -*- coding: utf-8 -*-
"""
Created on Wed Mar 21 19:39:26 2018

@author: Saul
"""

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack1 = []
        self.stack2 = []
    def push(self, node):
        # write code here
        self.stack1.insert(0,node)
    def pop(self):
        # return xx
        if len(self.stack2) != 0:
            return self.stack2.pop(0)
        elif len(self.stack1) == 0:
            return
        else:
            self.stack2 = self.stack1[::-1]
            self.stack1 = []
            return self.stack2.pop(0)