代码:
- 丑数 * 2 ———— 丑数 * 3 ———— 丑数 * 5 依旧是丑数,且这样运算也包含所有丑数
- 但为了避免丑数重复,所以在记录最小之后,如果其他运算得到的答案一样,则将其他的 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
}