面试题49:丑数
题目:我们把只包含因子2,3,5的数称作丑数(Ugly Number)。求按从小到大的顺序的第1500个丑数。例如,6,8都是丑数,但14不是,因为它包含因子7。习惯上我们把1当作第一个丑数。
思路1:暴力解法
从 1 开始往后加,判断每一个数是不是只有因子 2,3,5。每一个符合条件的给它计数,记到1500个即可。
代码实现:
public static int solve(int n) {
int count = 0;
int number = 1;
int temp = number;
while(count < n) {
while(temp != 1 && temp % 2 == 0) temp /= 2;
while(temp != 1 && temp % 3 == 0) temp /= 3;
while(temp != 1 && temp % 5 == 0) temp /= 5;
if(temp == 1) count++;
temp = ++number;
}
return number-1;
}
思路2:
发现!!每一个丑数都是前面某一个丑数的两倍、三倍或五倍。所以定义三个整数 p2, p3, p5。每次取 dp[p2] × 2,dp[p3] × 3,dp[p5] × 5 的最小值。为了去重,判断 dp[i] 与哪个值相同,比如与 dp[p2] × 2 相同,则 p2++。
代码实现:
public static int solve2(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++) {
dp[i] = Math.min(Math.min(dp[p2] * 2, dp[p3] * 3), dp[p5] * 5);
if (dp[i] == dp[p2] * 2) p2++;
if (dp[i] == dp[p3] * 3) p3++;
if (dp[i] == dp[p5] * 5) p5++;
}
return dp[n];
}