解法:
思想:
●丑数:
一个丑数乘以2,3,5之后,一定还是一个丑数。
●丑数序列:
计算丑数序列其实就是合并三个有序序列。
//即丑数序列乘以2,3,5
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得到三个序列,这三个序列合并得到丑数序列
所以可以通过一个1,来获得{2, 3, 5},然后将2加入序列,又可以通过2得到{4,6,10},然后将3加入序列…
●如何合并
合并有序序列的常用做法,通过三个指针,取最小的值
注意:可能会出现重复的,所以遇到相同的都++
if(dp[i] == dp[p2] * 2)
p2++;
if(dp[i] == dp[p3] * 3)
p3++;
if(dp[i] == dp[p5] * 5)
p5++;
复杂度:
●时间:O(N)
●空间:O(N),长度为N的dp
数组
代码:
class Solution {
public:
int nthUglyNumber(int n) {
if(n <= 0)
return -1;
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];
}
};