[路飞]_js算法：leetcode 373-查找和最小的 K 对数字

leetcode 373. 查找和最小的 K 对数字

问题描述： 给定两个以 升序排列 的整数数组 nums1 和 ****nums2 ****, 以及一个整数 k ****。

定义一对值 (u,v)，其中第一个元素来自 nums1，第二个元素来自 nums2 ****。

请找到和最小的 k 个数对 (u1,v1),  (u2,v2)  ...  (uk,vk) 。(by leetcode 373)

示例 1:

输入: nums1 = [1,7,11], nums2 = [2,4,6], k = 3
输出: [1,2],[1,4],[1,6]
解释: 返回序列中的前 3 对数：
     [1,2],[1,4],[1,6],[7,2],[7,4],[11,2],[7,6],[11,4],[11,6]

示例 2:

输入: nums1 = [1,1,2], nums2 = [1,2,3], k = 2
输出: [1,1],[1,1]
解释: 返回序列中的前 2 对数：
     [1,1],[1,1],[1,2],[2,1],[1,2],[2,2],[1,3],[1,3],[2,3]

思路： 堆实现 大顶堆

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @param {number} k
 * @return {number[][]}
 */
 var kSmallestPairs = function (nums1, nums2, k) {
    const heap=new Heap((a,b)=>a[0]+a[1]>b[0]+b[1]);
    for(const x of nums1){
        for(const y of nums2){
            let item=[x,y];
            heap.push(item);
            if(heap.size>k){
            //说明较小的值对已经够数了 不需要再往下找了
                if(heap.pop()===item)break;
            }
        }
    }
    return heap.getData()
  };

class Heap {
  constructor(cmp) {
      this.data = [];
      this.cmp = cmp;
  }
  get size() {
      return this.data.length;
  }
  get top() {
      return this.data[0];
  }
  getData() {
      return this.data;
  }
  swap(i, j) {
      [this.data[i], this.data[j]] = [this.data[j], this.data[i]];
  }
  // 向上冒泡
  up(i) {
      let index=this.data.length-1;
      while(index>0){
          let p=Math.floor((index-1)/2);
          if(p>=0&&this.cmp(this.data[index],this.data[p])){
              this.swap(index,p);
              index=p;
          }else{
              break;
          }
      }
  }
  // 下沉操作
  down(i) {
    if(this.data.length<2)return;
    let index=0,l=2*index+1,len=this.data.length;
    while(l<len){
      let r=l+1;
      if(r<len&&this.cmp(this.data[r], this.data[l]))l=r;
      if(this.cmp(this.data[index], this.data[l]))break;
      this.swap(index,l)
      index=l;
      l=index*2+1;
    }
  }
  push(item) {
    this.data.push(item);
    this.up();
  }
  //删除堆顶元素
  pop() {
      this.swap(0, this.data.length - 1);
      const res = this.data.pop();//已删除的元素（原来的堆顶元素）
      this.down();
      return res;
  }
}