题目:
我们把只包含因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。例如6、8都是丑数,但14不是,因为他包含因子7。习惯上我们把1当做第一个丑数。
解析:
普通方法例如依次判断每一个数是否为丑数,在进行统计,这样时间复杂度很高。因为每一个丑数都是要乘2,3或5。这时候为了避免重复判断丑数,则可以利用动态规划解决此问题,这样就能把时间复杂度降低到O(n)。
参考答案:
class Solution{
public:
int nthUglyNumber(int n){
if(n < 7)
return n;
vector<int> dp(n, 0);
dp[0] = 1;
int p2 = 0, p3 = 0, p5 = 0;
for(int i = 1; i < n; ++i){
dp[i] = min(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 - 1];
}
};