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

239.滑动窗口最大值

题目

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

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

进阶:

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

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

  • 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;
}; 

你可能感兴趣的:(javascript,算法,c++)