代码随想录 Day10 栈与队列

---

232. 用栈实现队列

用栈实现队列

class MyQueue(object):

    def __init__(self):
        self.stack1 = []
        self.stack2 = []

    def push(self, x):
        self.stack1.append(x)

    def pop(self):
        if not self.stack2:
            while self.stack1:
                self.stack2.append(self.stack1.pop())#pop为数组的指令
        return self.stack2.pop()

    def peek(self):
        if not self.stack2:
            while self.stack1:
                self.stack2.append(self.stack1.pop())
        return self.stack2[-1]

    def empty(self):
        return not self.stack1 and not self.stack2

加入stack_in为[1,2] 则 stack_out为[2,1]。

225. 用队列实现栈

用队列实现栈

 使用一个队列实现栈。

class MyStack:

    def __init__(self):
        self.queue=deque()

    def push(self, x: int) -> None:
        self.queue.append(x)

    def pop(self) -> int:
        if self.empty():
            return None
        for i in range(len(self.queue)-1):
            self.queue.append(self.queue.popleft())
        return self.queue.popleft()


    def top(self) -> int:
        if self.empty():
            return None
        return self.queue[-1]

    def empty(self) -> bool:
        return not self.queue

注意点

1. 首先定义一个队列，python自带的队列，deque().
2. 只有队列才有popleft操作。
3. 记得加self.queue。

补充

定义一个栈

 可以看出使用数组 [] 来表示的。

class Stack:
    def __init__(self):
        self.stack = []
 
    def push(self, element):
        self.stack.append(element)
 
    def pop(self):
        return self.stack.pop()
 
    def get_top(self):
        if len(self.stack) > 0:
            return self.stack[-1]   # 取栈顶元素即列表最后一个元素
        else:
            return None
 
    def is_empty(self):
        return len(self.stack) == 0
 
    def stack_all(self):
        return self.stack

定义队列

from collections import deque
 
# 单向队列
q = deque()
q2 = deque([1, 2, 3], 5)   # 第一个参数为初始进队元素，第二个元素设置队列最大长度
# 队满之后再进队时，队首元素会自动出队
# 对尾进队
q.append(1)
# 队首出队
print(q.popleft())
 
 
# 双向队列
q.appendleft(1)  # 队首进队
q.pop()  # 队尾出队