当前位置:   article > 正文

算法打卡day11

算法打卡day11

今日任务:

1)239. 滑动窗口最大值

2)347.前 K 个高频元素

239. 滑动窗口最大值

题目链接: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)

  1. class Solution:
  2. # 暴力解法,超时,时间复杂度O(kn)
  3. def maxSlidingWindow(self, nums: list[int], k: int) -> list[int]:
  4. left = 0
  5. right = k
  6. res = []
  7. while right <= len(nums):
  8. maxSum = max(nums[left:right])
  9. res.append(maxSum)
  10. left += 1
  11. right += 1
  12. return res

方法二:采用单调队列

1)首先,我们需要自己实现单调队列,队列中只维护最大值即可。队列是先进先出,那我们队列长度控制为k。

2)当进来一个数时,比较其与队列中最后一个数的大小

     若新增数大,则弹出队列中的的数。

        若队列中的数大,则添加新增数

        

3)当队列中满k个元素时,我们需要将最前面的元素弹出,同时新增一个数,重复2)过程

  1. class Solution:
  2. def maxSlidingWindow2(self, nums: list[int], k: int) -> list[int]:
  3. q = MyQueue()
  4. res = []
  5. # 将前k个元素放进队列
  6. for i in nums[:k]:
  7. q.push(i)
  8. # 收集最大值
  9. res.append(q.getMax())
  10. for i in range(k,len(nums)):
  11. # 移除窗口最前面的元素
  12. q.pop(nums[i-k])
  13. # 新增元素
  14. q.push(nums[i])
  15. # 获得当前最大元素,将最大值添加到列表中
  16. res.append(q.getMax())
  17. return res
  18. # 定义一个单调队列
  19. class MyQueue():
  20. def __init__(self):
  21. # 使用deque实现单调队列
  22. self.queue = deque()
  23. def pop(self,value):
  24. if self.queue and value == self.queue[0]:
  25. self.queue.popleft()
  26. def push(self,value):
  27. while self.queue and value > self.queue[-1]:
  28. self.queue.pop()
  29. self.queue.append(value)
  30. def getMax(self):
  31. return self.queue[0]

感想:

这题核心是要用单调队列结构。如果不熟悉这个结构就比较难。所以还是的反复做题,多熟悉数据结构

347.前 K 个高频元素

题目链接: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个元素,有一点要注意,应该采用小顶堆实现,每次弹出堆顶的最小数,把大数留下来

  1. import heapq
  2. class Solution:
  3. def topKFrequent(self, nums: list[int], k: int) -> list[int]:
  4. # 统计频率
  5. f = {}
  6. for i in nums:
  7. f[i] = f.get(i,0) + 1
  8. # 定义一个小顶堆,大小为k
  9. priority_queue = []
  10. for key,freq in f.items():
  11. heapq.heappush(priority_queue, (freq, key))
  12. if len(priority_queue) > k: # 如果堆的大小大于了K,则队列弹出,保证堆的大小一直为k
  13. heapq.heappop(priority_queue)
  14. # 找出前K个高频元素,因为小顶堆先弹出的是最小的,所以倒序来输出到数组
  15. result = [0] * k
  16. for i in range(k - 1, -1, -1):
  17. # 将弹出小顶堆中第二个值key存放在列表中
  18. result[i] = heapq.heappop(priority_queue)[1]
  19. return result

感想:

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/384927
推荐阅读
相关标签
  

闽ICP备14008679号