代码随想录二刷-队列及其应用题目(JS)【重要】
239.滑动窗口最大值
题目
给定一个数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回滑动窗口中的最大值。
进阶:
你能在线性时间复杂度内解决此题吗?
提示:
- 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 个高频元素的集合是唯一的。
- 你可以按任意顺序返回答案。
方法
思路很简单,大体分为三步
- 要统计元素出现频率(Map)
- 对频率排序(对频率进行排序,这里我们可以使用一种 容器适配器就是优先级队列)
- 找出前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;
};