[JavaScript 刷题] 堆 - 数据流中的第 K 大元素, leetcode 703

Posted GoldenaArcher

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[JavaScript 刷题] 堆 - 数据流中的第 K 大元素, leetcode 703相关的知识,希望对你有一定的参考价值。

[javascript 刷题] 堆 - 数据流中的第 K 大元素, leetcode 703

github repo 地址: https://github.com/GoldenaArcher/js_leetcode,Github 的目录 大概 会更新的更勤快一些。

题目地址:703. Kth Largest Element in a Stream

题目

如下:

Design a class to find the k^th largest element in a stream. Note that it is the k^th largest element in the sorted order, not the k^th distinct element.

Implement KthLargest class:

  • KthLargest(int k, int[] nums) Initializes the object with the integer k and the stream of integers nums.
  • int add(int val) Appends the integer val to the stream and returns the element representing the k^th largest element in the stream.

解题思路

整体来说没有什么特别难的地方,用 Min Heap 或者 BST 去解这道题都可以,不过因为是 leectode 上出现的这道题,最近又发现了一个 leetcode 中已经实现的 JavaScript 特有的 Heap 类,因此就在这里放一下了。

只在 leetcode 上刷题的,可以看一下这个类:MaxPriorityQueue 以及对应的 MinPriorityQueue,目前看到实现的功能有:

  • constructor

    constructor 的用法如下:

    this.pq = new MaxPriorityQueue( priority: (x) => -x );
    

    其中的 priority 就是一个 comparator,这样的话保存一些特殊的结点也是可以实现的。

  • enqueue

  • dequeue

  • front

  • back

使用 JavaScript 解题

/**
 * @param number k
 * @param number[] nums
 */
var KthLargest = function (k, nums) 
  this.pq = new MaxPriorityQueue( priority: (x) => -x );
  this.size = k;

  for (const num of nums) 
    this.pq.enqueue(num);

    if (this.pq.size() > this.size) 
      // console.log(this.pq.dequeue());
      this.pq.dequeue();
    
  
;

/**
 * @param number val
 * @return number
 */
KthLargest.prototype.add = function (val) 
  this.pq.enqueue(val);
  if (this.pq.size() > this.size) 
    this.pq.dequeue();
    // console.log(this.pq.dequeue());
  
  return this.pq.front().element;
;

/**
 * Your KthLargest object will be instantiated and called as such:
 * var obj = new KthLargest(k, nums)
 * var param_1 = obj.add(val)
 */

以上是关于[JavaScript 刷题] 堆 - 数据流中的第 K 大元素, leetcode 703的主要内容,如果未能解决你的问题,请参考以下文章

[JavaScript 刷题] 数组 - 替换数组中的元素,leetcode 2295

[JavaScript 刷题] 树 - 二叉搜索树中的插入操作, leetcode 701

[JavaScript 刷题] 树 - 二叉搜索树中的插入操作, leetcode 701

LeetCode刷题(130)~前 K 个高频元素最小堆|快速排列 难点!

[JavaScript 刷题] 树 - 删除二叉搜索树中的节点, leetcode 450

[JavaScript 刷题] 树 - 删除二叉搜索树中的节点, leetcode 450