两个堆栈实现队列/两个队列实现堆栈(Python)

堆栈与队列相互实现

两个堆栈实现队列

执行push操作时，将元素压入stack1中

执行pop操作时，若stack2 不空，则出栈顶元素

若stack2为空，则stack1逐个弹出元素并压入stack2（便满足了队列先进先出），然后stack2出栈顶元素

class Queue():
    def __init__(self):
        self.stack1 = [] # 入栈
        self.stack2 = [] # 出栈

    def push(self, value):
        self.stack1.append(value)
    
    def pop(self):
        if not self.stack1 and not self.stack2:#若两个堆栈皆空
            return None
            
        if self.stack2: # 若出栈不空
            return self.stack2.pop()
        
        # 若出栈未空，将stack1元素压入stack2
        while(self.stack1):
            self.stack2.append(self.stack1.pop())
        
        
        return self.stack2.pop()
        
ob = Queue()
ob.push(1)
ob.push(2)
print(ob.pop())
ob.push(3)
print(ob.pop())

两个队列实现堆栈

执行push操作时，将元素填入非空的那一个队列，若都空，任一插入

执行pop操作时，将非空队列，除了尾部元素(最后一个元素)的其他元素皆压入另外一个队列，然后将尾部元素弹出

class Stack():
    def __init__(self):
        self.queue1 = []
        self.queue2 = []
        
    def push(self,value):
        if self.queue1: # 若队列1 不空
            self.queue1.append(value)
        elif self.queue2: # 若队列2 不空
            self.queue2.append(value)
        else: # 两个队列皆空
            self.queue1.append(value)
            
    def pop(self):
        if not self.queue1 and not self.queue2:#若两个队列皆空
            return None
        if self.queue1: # 若队列1 非空
            
            while(len(self.queue1) > 1):
                self.queue2.append(self.queue1.pop(0))
            return self.queue1.pop()
        else: # 若队列2 非空
            while(len(self.queue2) > 1):
                self.queue1.append(self.queue2.pop(0))
            return self.queue2.pop()

ob = Stack()
ob.push(1)
ob.push(2)
print(ob.pop())
ob.push(3)
print(ob.pop())
print(ob.pop())
print(ob.pop())