题目介绍

264. 丑数 II

难度中等842收藏分享切换为英文接收动态反馈

给你一个整数 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 通常被视为丑数。

解题思路

1. 维护一个最小堆
2. 定义一个变量ans记录第n个丑数的值，并压入最小堆中
3. while循环，直到n === 0， 每次弹出一个最小堆的值，并赋值给ans，如果ans 可以被5整除，最小堆压入 ans * 5， 可以被 3 整除 压入ans * 5， ans* 3， 否则ans * 5， ans* 3， ans * 2
4. 循环结束返回ans

代码

/**
 * @param {number} n
 * @return {number}
 */
class Heap {
  constructor(heapType) {
    // 0是小顶堆，1是大顶堆
    this.heapType = heapType || 0
    this.data = []
  }
  // getTep(ind) {}
  // 向下调
  shiftDown(ind = 0) {
    let n = this.size() - 1
    while (ind * 2 + 1 <= n) {
      let tmp = ind
      // 小顶堆是用大于号 this.data[tmp] > this.data[ind * 2 + 1]， 大顶堆小于号 this.data[tmp] < this.data[ind * 2 + 1]
      if (this.heapType === 0) {
        if (this.data[tmp] > this.data[ind * 2 + 1]) tmp = ind * 2 + 1
        if (ind * 2 + 2 <= n && this.data[tmp] > this.data[ind * 2 + 2])
          tmp = ind * 2 + 2
      } else {
        if (this.data[tmp] < this.data[ind * 2 + 1]) tmp = ind * 2 + 1
        if (ind * 2 + 2 <= n && this.data[tmp] < this.data[ind * 2 + 2])
          tmp = ind * 2 + 2
      }
      if (tmp === ind) break
      this.data[tmp] = [this.data[ind], (this.data[ind] = this.data[tmp])][0]
      ind = tmp
    }
  }
  // 向上调整
  shiftUp(idx) {
    let pIdx = null
    if (this.heapType === 0) {
      while (
        ((pIdx = Math.floor((idx - 1) / 2)),
        // 小顶堆是用大于号 this.data[pIdx] > this.data[idx], 大顶堆用小于号this.data[pIdx] < this.data[idx]
        idx && this.data[pIdx] > this.data[idx])
      ) {
        this.data[pIdx] = [this.data[idx], (this.data[idx] = this.data[pIdx])][0]
        idx = pIdx
      }
    } else {
      while (
        ((pIdx = Math.floor((idx - 1) / 2)),
        // 小顶堆是用大于号 this.data[pIdx] > this.data[idx], 大顶堆用小于号this.data[pIdx] < this.data[idx]
        idx && this.data[pIdx] < this.data[idx])
      ) {
        this.data[pIdx] = [this.data[idx], (this.data[idx] = this.data[pIdx])][0]
        idx = pIdx
      }
    }
    
  }
  push(val) {
    this.data.push(val)
    this.shiftUp(this.size() - 1)
  }
  pop() {
    if (this.size() === 0) return
    if (this.size() === 1) {
      return this.data.pop()
    }
    this.data[0] = this.data.pop()
    this.shiftDown(0)
  }
  top() {
    return this.data[0]
  }
  size() {
    return this.data.length
  }
} 
var nthUglyNumber = function(n) {
    let h = new Heap(0)
    let ans = 1
    h.push(1)
    while(n--) {
        ans = h.top()
        h.pop()
        if (ans % 5 === 0) {
            h.push(ans * 5) 
        } else if (ans % 3 === 0) {
            h.push(ans * 3)
            h.push(ans * 5)
        } else {
            h.push(ans * 3)
            h.push(ans * 5)
            h.push(ans * 2)
        }
    }
    return ans
};