求解第n个丑数a[n],前面肯定存在三个最小的下标i,j,k属于(1,2,…,n-1)使得
a[i]*2,a[j]*3,a[k]*5刚好大于a[n-1],然后选取三者中最小的即为a[n]
下面看看这三个下标,它满足两个条件:1.乘以对应的因子一定大于a[n-1];2.满足1的同时又是最小的。
从头开始,a[1]=1,毫无疑问,i=j=k=1。显然,这时肯定选i,a[2]=2。现在2已经在数组a里了,下次
再选i的话就不满足条件1了,所以要对i更新i++,条件2是通过每次必须(更新)时只加1实现的。
同理,j和k的更新也跟i一样。不过有个问题需要注意,当a[n]=6时,a[3]*2可以得到,a[2]*3也可以得到
所以,i和j都要加1。即不能用if…else…这种结构,多个同级if才满足要求。
class Solution {
public:
int nthUglyNumber(int n) {
vector<int> dp(n, 1);
int p2 = 0;
int p3 = 0;
int p5 = 0;
for(int i = 1; i < n; i++)
{
//dp[i] = min(dp[p2] * 2, min(dp[p3] * 3, dp[p5]*5));
dp[i] = min(dp[p2]*2, dp[p3]*3);
dp[i] = min(dp[i], 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];
}
};
作者:he-he-28
链接:https://leetcode-cn.com/problems/ugly-number-ii/solution/dong-tai-gui-hua-xiang-jie-by-he-he-28/
来源:力扣(LeetCode)