代码随想录算法训练营第十三日| LC239. 滑动窗口最大值 LC347.前 K 个高频元素

LC239.滑动窗口最大值：

from collections import deque

class MyQueue:
    def __init__(self):
        self.queue = deque()
    
    def pop(self, value):
        if self.queue and value == self.queue[0]:
           self.queue.popleft()

    def push(self, value):
        while self.queue and value > self.queue[-1]:
            self.queue.pop()
        self.queue.append(value)

    def front(self):
        return self.queue[0]


class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        que = MyQueue()
        result = []
        for i in range(k):
            que.push(nums[i])
        result.append(que.front())

        for i in range(k, len(nums)):
            que.pop(nums[i - k])  #滑动窗口移除最前面元素
            que.push(nums[i])     #滑动窗口前加入最后面的元素
            result.append(que.front())  #记录对应的最大值
        
        return result

LC347.前 K 个高频元素

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        count = {}
        freq = [[] for i in range(len(nums) + 1)]

        for n in nums:
            count[n] = 1 + count.get(n, 0)
        for n, c in count.items():
            freq[c].append(n)
        
        res = []
        for i in range(len(freq) -1, 0 , -1):
            for n in freq[i]:
                res.append(n)
            if len(res) == k:
                return res