> 文章列表 > 代码随想录二刷-队列及其应用题目(JS)【重要】

代码随想录二刷-队列及其应用题目(JS)【重要】

代码随想录二刷-队列及其应用题目(JS)【重要】

239.滑动窗口最大值

题目

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

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

进阶:

你能在线性时间复杂度内解决此题吗?

代码随想录二刷-队列及其应用题目(JS)【重要】
提示:

  • 1 <= nums.length <= 10^5
  • -10^4 <= nums[i] <= 10^4
  • 1 <= k <= nums.length

方法

队列没有必要维护窗口里的所有元素,只需要维护有可能成为窗口里最大值的元素就可以了,同时保证队列里的元素数值是由大到小的。
这个维护元素单调递减的队列就叫做单调队列,即单调递减或单调递增的队列。
1.定义一个从队头到队尾递减的队列
2.输入k个元素,如果不满足递减原则则不加入队列中
3.每移动一个单位,如果滑动窗口移除的元素在队列中,则移除,不在队列中就不管了,然后加入元素看是否满足队列的条件,如果满足,则加入,否则不加入,队头元素即为所求(不弹出)
【有机会做一下单调栈】

代码

/* @param {number[]} nums* @param {number} k* @return {number[]}*/
var maxSlidingWindow = function(nums, k) {class monoQueue {queue;constructor() {this.queue = [];}enqueue(val) {let back = this.queue[this.queue.length - 1];while (this.queue.length && back < val) {this.queue.pop();// while循环时一定不要忘记更新值back = this.queue[this.queue.length - 1]}this.queue.push(val);}dequeue(val) {// this.front() 是定义数据结构的方法// this.queue.pop() 是queue数组本身自带的方法if (this.front() == val) {this.queue.shift();}}front() {return this.queue[0];}}let helpQueue = new monoQueue();let resArr = [];for (let i = 0;i < k;i++) {helpQueue.enqueue(nums[i]);}resArr.push(helpQueue.front());for (let i = k;i < nums.length;i++) {helpQueue.enqueue(nums[i]);helpQueue.dequeue(nums[i - k]);resArr.push(helpQueue.front());}return resArr;
};

347.前k个高频元素

题目

给定一个非空的整数数组,返回其中出现频率前 k 高的元素。

示例 1:

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

示例 2:

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

提示:

  • 你可以假设给定的 k 总是合理的,且 1 ≤ k ≤ 数组中不相同的元素的个数。
  • 你的算法的时间复杂度必须优于 O ( n log ⁡ n ) O(n \\log n) O(nlogn) , n 是数组的大小。
  • 题目数据保证答案唯一,换句话说,数组中前 k 个高频元素的集合是唯一的。
  • 你可以按任意顺序返回答案。

方法

思路很简单,大体分为三步

  1. 要统计元素出现频率(Map)
  2. 对频率排序(对频率进行排序,这里我们可以使用一种 容器适配器就是优先级队列)
  3. 找出前K个高频元素

优先级队列是一个披着队列外衣的堆,因为优先级队列对外接口只是从对头取元素,从队尾添加元素,再无其他取元素的方式,看起来就是一个队列

堆是一棵完全二叉树,树中每个结点的值都不小于(或不大于)其左右孩子的值。如果父亲结点大于等于左右孩子就是大顶堆,小于等于左右孩子就是小顶堆

使用大顶堆就要把所有元素都进行排序,那能不能只排序k个元素呢?

所以我们要用小顶堆,因为要统计最大前k个元素,只有小顶堆每次将最小的元素弹出,最后小顶堆里积累的才是前k个最大元素。

JS语言中没有堆这种数据结构,我们需要自己定义

排序算法(大顶堆还是小顶堆)

从队尾插入一个元素

删除堆顶元素

在本科时学了很多堆的理论,但之前用Java刷题时并没有涉及,现在给你一个实现那些理论的机会

Java直接给你定义好了

代码

/* @param {number[]} nums* @param {number} k* @return {number[]}*/
class Heap {constructor(compareFn) {this.compareFn = compareFn;this.queue = [];}push(val) {this.queue.push(val);let index = this.queue.length - 1;let parent = Math.floor((index - 1) / 2);while (parent >= 0 && this.compare(parent,index) > 0) {[this.queue[index],this.queue[parent]] = [this.queue[parent],this.queue[index]];//更新下标index = parent;parent = Math.floor((index - 1) / 2);}}pop() {const out = this.queue[0];this.queue[0] = this.queue.pop();// 下沉let index = 0;let left = 1;// 选择左右两个子节点中小的那个(针对小顶堆而言)let searchChild = this.compare(left,left + 1) > 0 ? left + 1 : left;while (searchChild != undefined && this.compare(index,searchChild) > 0) {[this.queue[index],this.queue[searchChild]] = [this.queue[searchChild],this.queue[index]];//更新下标index = searchChild;left = 2 * index + 1;searchChild = this.compare(left,left + 1) > 0 ? left + 1 : left;}return out;}size() {return this.queue.length;}compare(index1,index2) {if (this.queue[index1] == undefined)    return 1;if (this.queue[index2] == undefined)    return -1;return this.compareFn(this.queue[index1],this.queue[index2]);}
}
var topKFrequent = function(nums, k) {let heap = new Heap((a,b) => a[1] - b[1]);let map = new Map();for (let i = 0;i < nums.length;i++) {map.set(nums[i],(map.get(nums[i]) || 0) + 1);}for (const entry of map.entries()) {heap.push(entry);if (heap.size() > k) {heap.pop();}}let resArr = [];for (let i = k - 1;i >= 0;i--) {resArr[i] = heap.pop()[0];}return resArr;
};