跟着leedcode刷算法 -- 堆、栈与队列3

130 阅读1分钟

这是我参与11月更文挑战的第16天,活动详情查看:2021最后一次更文挑战

image.png

题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

执行结果:

image.png

题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

相关标签

  • 队列
  • 数组
  • 滑动窗口
  • 单调队列
  • 堆(优先队列)

第一时间想到的方法 然而超时了

image.png

学习了一下使用双端队列去实现

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

执行结果:

image.png