LeetCode263. 丑数
https://leetcode-cn.com/problems/ugly-number/
编写一个程序判断给定的数是否为丑数。
丑数就是只包含质因数 2, 3, 5
的正整数。
输入: 6
输出: true
解释: 6 = 2 × 3
思路
首先判断特殊情况,然后判断num是否能被2或3或5整除,可以整除就用num除以2或3或5,不可以整除就退出循环,最后判断最终结果是否为1,是1为丑数,不是1不为丑数。本题十分简单,仅作为下一题的铺垫
代码
class Solution
{
public:
bool isUgly(int num)
{
if (num <= 0)
{
return false;
}
if (num == 1)
{
return true;
}
while (true)
{
if (num % 2 ==0) num = num / 2;
else if (num % 3 == 0) num = num / 3;
else if (num % 5 == 0) num = num / 5;
else break;
}
if (num == 1)
{
return true;
}
return false;
}
};
LeetCode264. 丑数II
编写一个程序,找出第 n
个丑数。
丑数就是质因数只包含 2, 3, 5
的正整数。
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
思路
1.思考状态:找出最优解的性质,明确状态表示什么
dp[i] 表示第i个丑数的值
2.思考状态转移方程:大问题的最有解如何由小问题的最优解得到
首先明确:对于每个丑数来说,它一定是从某个丑数乘以2或3或5得到的。
我们定义三个指针p2、p3、p5,初值都为1,这三个指针分别指向即将被2或3或5乘以的数值
所以:dp[i] = min(dp[p2] * 2, dp[p3] * 3, dp[p5] * 5)
当dp[i] 更新为三者中的最小值后,说明三个指针p2、p3、p5中的某个指针当前工作已完成,于是将其指向下一个数值(即指针所指的数值加1)
3.思考初始状态
1为第一个丑数,所以:dp[1] = 1
4.自底向上计算得到最优解
5.思考是否可以进行空间的优化
代码
class Solution
{
public:
//动态规划
int nthUglyNumber(int n)
{
vector<long long> dp(n + 1, 0);
dp[1] = 1;
int p2 = 1, p3 = 1, p5 = 1;
for (int i = 2; 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];
}
};