丑数
1.题目描述
编写一个程序,找出第 n 个丑数。丑数就是只包含质因数 2, 3, 5 的正整数。
示例:
说明:
1.1 是丑数。
2.n 不超过1690。
2.思路
动态规划:丑数是另外一个丑数乘以2,3,5的结果,从1开始,按递增顺序求出丑数。动态方程dp[i]=min(dp[p2]*2,dp[p3]*3,dp[p5]*5)
1.根据动态方程,求出大于当前丑数的最小丑数newNum。
2.判断newNum是由哪个数得到的,即更新newNum的下标p2,p3,p5。
3.当数组的大小等于n时,返回newNum就是第n个丑数。
3.代码
class Solution {
public:
int nthUglyNumber(int n) {
if(n < 7){
return n;
}
int p2 = 0, p3 = 0, p5 = 0;
int newNum = 1;
vector<int> vec;
vec.push_back(newNum);
//丑数是另外一个丑数乘以2,3,5的结果
while(vec.size() < n){
newNum = min(vec[p2]*2,min(vec[p3]*3,vec[p5]*5));//得出大于当前丑数的最小的丑数
//得到newNum的下标
if(vec[p2]*2 == newNum){//1,2,3,4,5,6,
p2++;
}
if(vec[p3]*3 == newNum){
p3++;
}
if(vec[p5]*5 == newNum){
p5++;
}
vec.push_back(newNum);
}
return newNum;
}
};
4.复杂度分析
时间复杂度:O(n)
空间复杂度:O(n)