赞
踩
今日任务:
1)239. 滑动窗口最大值
2)347.前 K 个高频元素
题目链接:239. 滑动窗口最大值 - 力扣(LeetCode)
给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回滑动窗口中的最大值。示例 1:
输入: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
--------------------------------------------示例 2:
输入:nums = [1], k = 1
输出:[1]
文章讲解:代码随想录 (programmercarl.com)
视频讲解:单调队列正式登场!| LeetCode:239. 滑动窗口最大值哔哩哔哩bilibili
方法一:暴力解法思路
直接采用双指针循环遍历数组,每次取出滑动窗口中最大值,提交超时,时间复杂度为O(kn)
- class Solution:
- # 暴力解法,超时,时间复杂度O(kn)
- def maxSlidingWindow(self, nums: list[int], k: int) -> list[int]:
- left = 0
- right = k
- res = []
- while right <= len(nums):
- maxSum = max(nums[left:right])
- res.append(maxSum)
- left += 1
- right += 1
-
- return res
方法二:采用单调队列
1)首先,我们需要自己实现单调队列,队列中只维护最大值即可。队列是先进先出,那我们队列长度控制为k。
2)当进来一个数时,比较其与队列中最后一个数的大小
若新增数大,则弹出队列中的的数。
若队列中的数大,则添加新增数
3)当队列中满k个元素时,我们需要将最前面的元素弹出,同时新增一个数,重复2)过程
- class Solution:
- def maxSlidingWindow2(self, nums: list[int], k: int) -> list[int]:
- q = MyQueue()
- res = []
- # 将前k个元素放进队列
- for i in nums[:k]:
- q.push(i)
-
- # 收集最大值
- res.append(q.getMax())
-
- for i in range(k,len(nums)):
- # 移除窗口最前面的元素
- q.pop(nums[i-k])
-
- # 新增元素
- q.push(nums[i])
-
- # 获得当前最大元素,将最大值添加到列表中
- res.append(q.getMax())
-
- return res
-
-
-
- # 定义一个单调队列
- class MyQueue():
- def __init__(self):
- # 使用deque实现单调队列
- 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 getMax(self):
- return self.queue[0]

感想:
这题核心是要用单调队列结构。如果不熟悉这个结构就比较难。所以还是的反复做题,多熟悉数据结构
题目链接:347. 前 K 个高频元素 - 力扣(LeetCode)
给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]示例 2:
输入: nums = [1], k = 1
输出: [1]
文章讲解:代码随想录 (programmercarl.com)
视频讲解:优先级队列正式登场!大顶堆、小顶堆该怎么用?| LeetCode:347.前 K 个高频元素哔哩哔哩bilibili
思路:
1)首先先用map求频率
2)然后采用优先队列去对频率排序,维护前k个元素,有一点要注意,应该采用小顶堆实现,每次弹出堆顶的最小数,把大数留下来
- import heapq
-
- class Solution:
- def topKFrequent(self, nums: list[int], k: int) -> list[int]:
- # 统计频率
- f = {}
- for i in nums:
- f[i] = f.get(i,0) + 1
-
- # 定义一个小顶堆,大小为k
- priority_queue = []
- for key,freq in f.items():
- heapq.heappush(priority_queue, (freq, key))
- if len(priority_queue) > k: # 如果堆的大小大于了K,则队列弹出,保证堆的大小一直为k
- heapq.heappop(priority_queue)
-
- # 找出前K个高频元素,因为小顶堆先弹出的是最小的,所以倒序来输出到数组
- result = [0] * k
- for i in range(k - 1, -1, -1):
- # 将弹出小顶堆中第二个值key存放在列表中
- result[i] = heapq.heappop(priority_queue)[1]
- return result

感想:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。