Python基础编程练习LeetCode——队列

Python基础编程练习LeetCode——队列

Python基础编程练习LeetCode——队列

题目来源：LeetCode：27，225，933

27、数组移除元素

给定一个数组和一个值,在这个数组中原地移除指定值和返回移除后新的数组长度。

输入

[3,2,2,3]
3

输出

2

解题思路

解题思路：利用python数据结构中list的pop操作，从数组的最后一个元素开始向前遍历，遇到和要删掉的元素，就pop一个元素出来。

完整代码

class Solution:
    def removeElement(self, nums: List[int], val: int) -> int:
        j=len(nums)
        for i in range(j-1,-1,-1):
            if nums[i]==val:
                nums.pop(i)    
        return len(nums)

225、用队列实现栈

使用队列实现栈的下列操作：

• push(x) --元素 x 入栈
• pop() – 移除栈顶元素
• top() – 获取栈顶元素
• empty() --返回栈是否为空

输入输出

MyStack stack = new MyStack();

stack.push(1);
stack.push(2);
stack.top(); // returns 2
stack.pop(); // returns 2
stack.empty(); // returns false

解题思路

解题思路：用两个栈实现，queue1用来存放数据，每次入栈就插入queue1索引值为0的地方。queue2为出栈时的队列，将queue1[1:-1]的元素出队然后入队queue2 (queue1逆序压入queue2)，然后queue1仅剩队首元素，这个时候出队即出栈。

完整代码

class MyStack:

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.queue1 = []
        self.queue2 = []

    def push(self, x):
        """
        Push element x onto stack.
        :type x: int
        :rtype: void
        """
        self.queue1.insert(0, x)

    def pop(self):
        """
        Removes the element on top of the stack and returns that element.
        :rtype: int
        """
        for _ in range(len(self.queue1) - 1):
            self.queue2.insert(0, self.queue1.pop())
        res = self.queue1.pop()
        self.queue1 = self.queue2
        self.queue2 = []
        return res

    def top(self):
        """
        Get the top element.
        :rtype: int
        """
        self.queue2 = self.queue1[:]
        for i in range(len(self.queue1) - 1):
            self.queue1.pop()
        res = self.queue1[-1]
        self.queue1 = self.queue2
        self.queue2 = []
        return res

    def empty(self):
        """
        Returns whether the stack is empty.
        :rtype: bool
        """
        if len(self.queue1) == 0:
            return True
        return False

933、最近的请求次数

找出时间在3000内发出ping请求个数。每次ping参数t都是大于前面的。

输入

inputs = [“RecentCounter”,“ping”,“ping”,“ping”,“ping”], inputs = [[],[1],[100],[3001],[3002]]

输出

[null,1,2,3,3]

解题思路

解题思路：使用一个队列，当t时间到达之后，在t-3000之前的调用全部删除，然后求队列长度。

完整代码

class RecentCounter:

    def __init__(self):
        self.que = collections.deque()

    def ping(self, t):
        """
        :type t: int
        :rtype: int
        """
        while self.que and self.que[0] < t - 3000:
            self.que.popleft()
        self.que.append(t)
        return len(self.que)