代码随想录刷题第十三天 | 239. 滑动窗口最大值 ● 347.前 K 个高频元素

代码随想录刷题第十三天

滑动窗口最大值 (LC 239) 难！！！

给你一个整数数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回 滑动窗口中的最大值 。

示例：

输入：nums = [1,3,-1,-3,5,3,6,7], k = 3
输出：[3,3,5,5,6,7]
解释：
滑动窗口的位置 | 最大值
[1 3 -1] -3 5 3 6 7 | 3
1 [3 -1 -3] 5 3 6 7 | 3
1 3 [-1 -3 5] 3 6 7 | 5
1 3 -1 [-3 5 3] 6 7 | 5
1 3 -1 -3 [5 3 6] 7 | 6
1 3 -1 -3 5 [3 6 7] | 7

题目思路：

本题涉及手动编写一个python队列，实现队首记录最大值的功能

代码实现：

class Myque(object):
    def __init__(self):
        self.q = deque()

    def pop(self,value):
        if self.q and value == self.q[0]:
            self.q.popleft()
    
    def push(self, value):
        while self.q and value > self.q[-1]:
            self.q.pop()
        self.q.append(value)
    
    def getmax(self):
        return self.q[0]

class Solution(object):
    def maxSlidingWindow(self, nums, k):
        q = Myque()
        res = []
        for i in range(k):
            q.push(nums[i])
        res.append(q.getmax())

        for j in range(k,len(nums)):
            q.pop(nums[j-k])
            q.push(nums[j])            
            res.append(q.getmax())
        return res

Python堆的实现，最大堆最小堆

前k个高频元素 (LC 347)

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

示例：

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

输入: nums = [1], k = 1
输出: [1]

题目思路：

代码实现：

class Solution(object):
    def topKFrequent(self, nums, k):

        mapping = {}
        for num in nums:
            mapping[num] = mapping.get(num,0)+1

        minheap = []
        for key, freq in mapping.items():
            heapq.heappush(minheap, (freq, key))
            while len(minheap) > k:
                heapq.heappop(minheap)
        
        res = []

        for i in range(len(minheap)):
            res.append(minheap[i][1])
        return res