丑数——动态规划

54 阅读1分钟

image.png

代码:

  1. 丑数 * 2 ———— 丑数 * 3 ———— 丑数 * 5 依旧是丑数,且这样运算也包含所有丑数
  2. 但为了避免丑数重复,所以在记录最小之后,如果其他运算得到的答案一样,则将其他的 dp 下标也后推一次
func nthUglyNumber(n int) int {
    dp := make([]int, n+1)
    dp[1] = 1
    p2, p3, p5 := 1, 1, 1
    for i := 2; i <= n; i++ {
        x2, x3, x5 := dp[p2]*2, dp[p3]*3, dp[p5]*5
        dp[i] = min(x2, x3, x5)
        if dp[i] == x2 {
            p2++
        }
        if dp[i] == x3 {
            p3++
        }
        if dp[i] == x5 {
            p5++
        }
    }
    return dp[n]
}

func min(a, b, c int) int {
    if a <= b && a <= c {
        return a
    }
    if b <= a && b <= c {
        return b
    }
    return c
}