这是我参与11月更文挑战的第16天,活动详情查看:2021最后一次更文挑战
题5
前 K 个高频元素
给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。
示例 1:
- 输入: nums = [1,1,1,2,2,3], k = 2
- 输出: [1,2]
示例 2:
- 输入: nums = [1], k = 1
- 输出: [1]
提示:
-
1 <= nums.length <= 105
-
k 的取值范围是 [1, 数组中不相同的元素的个数]
-
题目数据保证答案唯一,换句话说,数组中前 k 个高频元素的集合是唯一的
-
进阶:你所设计算法的时间复杂度 必须 优于 O(n log n) ,其中 n 是数组大小。
相关标签
- 数组
- 哈希表
- 分治
- 桶排序
- 计数
- 快速选择
- 排序
- 堆(优先队列)
1.字典计数 2.对字典进行排序 倒排
class Solution:
def topKFrequent(self, nums: List[int], k: int) -> List[int]:
doc = {}
for row in nums:
doc.setdefault(row,0)
doc[row]+=1
res= sorted(doc.items(), key=lambda d: d[1],reverse=True)
response=[]
for row in range(k):
response.append(res[row][0])
return response
执行结果:
题6
滑动窗口最大值
给你一个整数数组 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] 示例 3:
-
输入:nums = [1,-1], k = 1
-
输出:[1,-1] 示例 4:
-
输入:nums = [9,11], k = 2
-
输出:[11] 示例 5:
-
输入:nums = [4,-2], k = 2
-
输出:[4]
提示:
- 1 <= nums.length <= 105
- -104 <= nums[i] <= 104
- 1 <= k <= nums.length
相关标签
- 队列
- 数组
- 滑动窗口
- 单调队列
- 堆(优先队列)
第一时间想到的方法 然而超时了
学习了一下使用双端队列去实现
class Solution:
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
deque = collections.deque()
res = []
for index, value in enumerate(nums):
if deque and deque[0] <= index-k:
deque.popleft()
while deque and nums[deque[-1]] < value:
deque.pop()
deque.append(index)
if index >= k-1:
res.append(nums[deque[0]])
return res
执行结果: