思路:每个丑数都是由前一个丑数 x2或x3或x5而来,那么就会
nums2 = {1*2, 2*2, 3*2, 4*2, 5*2, 6*2, 8*2...}
nums3 = {1*3, 2*3, 3*3, 4*3, 5*3, 6*3, 8*3...}
nums5 = {1*5, 2*5, 3*5, 4*5, 5*5, 6*5, 8*5...}
可以得到三个这样的数组,我们维护三个指针,每次都在三个数组中取出一个最小的。
核心思想:当一个丑数已经被* 2 * 3 * 5后,对于生成丑数已经没有用了,所以此时指针要向前移动。
注意点:三个并列的都是if
语句,所以出现相同值的时候指针都会移动,所以不会出现重复值。如果是if..else
会有重复。
// 1, 2, 3, 4, 5, 6, 8, 9, 10, 12
class Solution {
public static int nthUglyNumber(int n) {
int a = 0, b = 0, c = 0;
int[] dp = new int[n];
dp[0] = 1;
for (int i = 1; i < n; i++) {
int n2 = dp[a] * 2, n3 = dp[b] * 3, n5 = dp[c] * 5;
dp[i] = Math.min(Math.min(n2, n3), n5);
if (dp[i] == n2) a++;
if (dp[i] == n3) b++;
if (dp[i] == n5) c++;
}
return dp[n - 1];
}
public static void main(String[] args) {
int n = 10;
System.out.println(nthUglyNumber(n));
}