码迷,mamicode.com
首页 > 其他好文 > 详细

用两个栈实现队列

时间:2018-08-30 02:03:42      阅读:136      评论:0      收藏:0      [点我收藏+]

标签:div   一个栈   数据   两个栈   []   ret   完成   ini   一个   

题目描述

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

思路

首先要明确栈和队列的特性,栈是后入先出的数据类型,队列是先进先出的数据类型,那么就可以用两个栈,一个存储新的push进队列的数据,另一个栈用来pop出队头的数据,push很简单,直接利用栈的特性append进去就可以了,那么pop怎么办呢,需要把队头(也就是最先push的数据)pop出去。有一个办法,就是假如

1、栈2没有数据,而栈1有数据的话,那就把栈1的数据全pop到栈2去,然后直接用栈2pop

2、栈1和栈2都没有数据,那就返回None

3、栈2有数据的话,直接pop栈2的数据

这样能保证栈2的数据都是“老数据”,且栈2的尾数据一定是最先push到所谓队列的那个队头;而栈1的数据都是“新push的数据”

解答

class Solution:
    def __init__(self):
        self._stack1 = []
        self._stack2 = []
    def push(self, node):
        # write code here
        self._stack1.append(node)
    def pop(self):
        # return xx
        if self._stack2:
            return self._stack2.pop()
        elif not self._stack1:
            return None
        else:
            while self._stack1:
                self._stack2.append(self._stack1.pop())
            return self._stack2.pop()

 

用两个栈实现队列

标签:div   一个栈   数据   两个栈   []   ret   完成   ini   一个   

原文地址:https://www.cnblogs.com/yqpy/p/9557704.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!