【剑指offer】5.用两个栈来实现队列[by Python]

最新推荐文章于 2021-11-24 18:18:05 发布

转载最新推荐文章于 2021-11-24 18:18:05 发布 · 165 阅读

本文介绍了一种使用两个栈实现队列的方法，通过巧妙地利用栈的特性，实现了队列的先进先出特性。主要讨论了如何在两个栈之间转换元素以完成队列的Push和Pop操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

GitHub：https://github.com/cytues/sword

源码

题目描述

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


 
   
    
   
   
    
     # -*- coding:utf-8 -*-
    
   

   
    
   
   
    
     ''
     '
    
   

   
    
   
   
    
     A是入栈的，B是出栈的，入栈时，直接进入A即可，出栈时，先判断是否有元素，
    
   

   
    
   
   
    
     如果B没有元素，pop肯定报错，应该先将A中所有的元素压倒B里面，再pop最上面一个元素，如果B中有就直接pop出，就可以，
    
   

   
    
   
   
    
     这是最优的思路，两个栈实现了先进后出，在一起又实现了队列的先进先出。
    
   

   
    
   
   
    
     '
     ''
    
   

   
    
   
   
    
     class Solution:
    
   

   
    
   
   
        
     # 初始化两个栈
    
   

   
    
   
   
        
     def __init__(self):
    
   

   
    
   
   
            
     self.stackA = []
    
   

   
    
   
   
            
     self.stackB = []
    
   

   
    
   
   
        
     # 这里只要求实现队列的push和pop操作，分别使用两个栈表示弹出和压入
    
   

   
    
   
   
        
     def push(self, node):
    
   

   
    
   
   
        
     # write code here
    
   

   
    
   
   
            
     self.stackA.append(node)
    
   

   
    
   
   
        
     # 弹出需要有一个先验条件：若队列为空，则返回None
    
   

   
    
   
   
        
     def pop(self):
    
   

   
    
   
   
            
     # 若压入的栈中有元素则直接弹出
    
   

   
    
   
   
            
     if 
     self.
     stackB:
    
   

   
    
   
   
                
     self.stackB.pop()
    
   

   
    
   
   
            
     # 若A中没有元素则返回None
    
   

   
    
   
   
    
             elif 
     not 
     self.
     stackA:
    
   

   
    
   
   
                
     return None
    
   

   
    
   
   
            
     # 若A中有元素，则统一压入B中进行弹出
    
   

   
    
   
   
            
     else:
    
   

   
    
   
   
                
     while 
     self.
     stackA:
    
   

   
    
   
   
                    
     self.stackB.append(
     self.stackA.pop())
    
   

   
    
   
   
                
     return 
     self.stackB.pop()