题目:
编写一个程序,找出第 n 个丑数。
丑数就是质因数只包含 2, 3, 5 的正整数。
题解思路:
方法一:记忆化的迭代+自下而上
class Solution {
public:
int nthUglyNumber(int n) {
int a=1,b=1,c=1;
vector<int>nums(n+1,0);
nums[0]=0;
nums[1]=1;
for(int i=2;i<=n;i++)
{
nums[i]=min(2*nums[a],min(3*nums[b],5*nums[c]));
//int cur=nums[i];
if(2*nums[a]==nums[i])
{
a++;
}
if(3*nums[b]==nums[i])
{
b++;
}
if(5*nums[c]==nums[i])
{
c++;
}
}
return nums[n];
}
};
方法二:动态规划
函数代码:
class Solution {
public:
int nthUglyNumber(int n) {
vector<int>dp(n,0);
dp[0]=1;
int a=0,b=0,c=0;
for(int i=1;i<n;i++)
{
dp[i]=min(dp[a]*2,min(dp[b]*3,dp[c]*5));
if(dp[i]==dp[a]*2)
{
a++;
}
if(dp[i]==dp[b]*3)
{
b++;
}
if(dp[i]==dp[c]*5)
{
c++;
}
}
return dp[n-1];
}
};
函数代码二:
class Solution {
public:
int nthUglyNumber(int n) {
int a=1,b=1,c=1;
vector<int>nums(n+1,0);
nums[0]=0;
nums[1]=1;
for(int i=2;i<=n;i++)
{
int cur=min(2*nums[a],min(3*nums[b],5*nums[c]));
nums[i]=cur;
if(2*nums[a]==cur)
{
a++;
}
if(3*nums[b]==cur)
{
b++;
}
if(5*nums[c]==cur)
{
c++;
}
}
return nums[n];
}
};
函数代码三:
class Solution {
public:
int nthUglyNumber(int n) {
int a=1,b=1,c=1;
vector<int>dp(n+1,0);
dp[0]=0;
dp[1]=1;
vector<int>nums(n+1,0);
nums[0]=0;
nums[1]=1;
for(int i=2;i<=n;i++)
{
dp[i]=min(2*nums[a],min(3*nums[b],5*nums[c]));
int cur=dp[i];
nums[i]=cur;
if(2*nums[a]==dp[i])
{
a++;
}
if(3*nums[b]==dp[i])
{
b++;
}
if(5*nums[c]==dp[i])
{
c++;
}
}
return dp[n];
}
};