一、题目描述:264. 丑数 II(中等)
编写一个程序,找出第 n 个丑数。
丑数就是只包含质因数 2, 3, 5 的正整数。
示例:
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
说明:1 是丑数。
n 不超过1690。来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ugly-number-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解题思路
三指针动态规划法
1、定义三个指针i2,i3,i5.分别表示当前2的倍数的最小值位置,3的倍数的最小值位置,5的倍数最小值位置。
2、每次计算获取dp[i2]*2,dp[i3]*3,dp[i5]*5的最小值,放入队列末尾。如果是最小值是dp[i2]*2则i2++,如果是dp[i3]*3 i3++,如果是dp[i5]*5 i5++。这里为了避免重复,三个都要判断一下
直到队列总数为n跳出循环
另外还有一个方法就是,使用一个最小堆,每次将堆顶元素出堆,然后分别乘以2、3、5加入堆,直到取出的数量为n
三、代码(三指针法)
class Solution {
public:
int nthUglyNumber(int n) {
if(n < 0)
return 0;
vector<int> dp(1,1);
int i2 = 0,i3=0,i5=0,i=1;
while(i++ < n)
{
int tmp = min(min(dp[i2]*2,dp[i3]*3),dp[i5]*5);
if(tmp == dp[i2]*2)i2++;
if(tmp == dp[i3]*3)i3++;
if(tmp == dp[i5]*5)i5++;
dp.push_back(tmp);
//cout<< tmp<<endl;
}
//cout<<n<<endl;
return *dp.rbegin();
}
};