先给出一个比较笨的办法
通过打表的方式把所有的丑数都列出来,然后取第 n 个
所有的丑数都是 2n1 * 3n2 * 5n3 (其中n1,n2,n3 可以为0),仿照广搜的思想打表
class Solution {
public int nthUglyNumber(int n) {
List<Integer> l = new ArrayList<>();
Queue<Integer> q = new LinkedList<>();
q.offer(1);
while(!q.isEmpty() ){
List<Integer> t = new ArrayList<>();
for(int i = q.size(); i > 0; i --){
int m = q.poll();
if(m < Integer.MAX_VALUE / 2 && m % 3 != 0 && m % 5 != 0)
q.offer(m * 2);
if(m < Integer.MAX_VALUE / 3 && m % 5 != 0)
q.offer(m * 3);
if(m < Integer.MAX_VALUE / 5)
q.offer(m * 5);
t.add(m);
}
l.addAll(t);
}
Collections.sort(l,(o1,o2) -> o1 - o2);
return l.get(n - 1);
}
}
动态规划的思想
dp[n] 表示第 n 个丑数
三个指针,一个 *2 一个 *3 一个 *5,每次取较小的那个,保证dp 数组的递增
class Solution {
public int nthUglyNumber(int n) {
int[] dp = new int[n+1];
dp[1] = 1;
int p2 = 1, p3 = 1, p5 = 1;
for (int i = 2; i <= n; i++) {
int num2 = dp[p2] * 2, num3 = dp[p3] * 3, num5 = dp[p5] * 5;
dp[i] = Math.min(Math.min(num2, num3), num5);
if (dp[i] == num2) {
p2++;
}
if (dp[i] == num3) {
p3++;
}
if (dp[i] == num5) {
p5++;
}
}
return dp[n];
}
}