[路飞]_js算法：leetcode 264-丑数 II

leetcode 264. 丑数 II

问题描述： 给你一个整数 n ，请你找出并返回第 n 个 丑数 。

丑数 就是只包含质因数 2、3 和/或 5 的正整数。

示例 1：

输入： n = 10
输出： 12
解释： [1, 2, 3, 4, 5, 6, 8, 9, 10, 12] 是由前 10 个丑数组成的序列。

示例 2：

输入： n = 1
输出： 1
解释： 1 通常被视为丑数。

思路一： 指针

/**
 * @param {number} n
 * @return {number}
 */
var nthUglyNumber = function(n) {
    let arr=[1];
    let p2=0;
     let p3=0;
      let p5=0;
      while(arr.length<n){
          let v2=arr[p2]*2;
           let v3=arr[p3]*3;
            let v5=arr[p5]*5;
        let min=Math.min(v2,v3,v5);
        if(min!==arr[arr.length-1])arr.push(min);
            if(min==v2)p2+=1;
             if(min==v3)p3+=1;
              if(min==v5)p5+=1;
      }
      return arr[n-1]
};

思路二： 用堆的方法来做：

/**
 * @param {number} n
 * @return {number}
 */
var nthUglyNumber = function(n) {
    let heap=new Heap((a,b)=>a<b);
    heap.push(1);
    let res=0;
    while(n--){
      res=heap.top;
        heap.pop();
        if(res%5===0){
            heap.push(res*5)
        }else if(res%3===0){
            heap.push(res*3);
            heap.push(res*5)
        }else{
            heap.push(res*5);
            heap.push(res*3);
            heap.push(res*2)
        }
    }
    return res;
   
};
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;
  }
}