leetcode 264.丑数Ⅱ 动态规划
思路:丑数第一个是1,其他都可由前面得到的每个丑数分别乘以2,3,5得到, 关键是丑数序列的递增排序。选用3个指针指向已得到的丑数序列元素,每次新生成的丑数中选取最小的一个加入丑数序列,然后相应的指针后移一位,这样得到的丑数序列就是递增的。
class Solution {
public:
int nthUglyNumber(int n) {
if(n == 0)
return 0;
if(n == 1)
return 1;
vector<int>dp;
dp.push_back(1);
int p2 = 0, p3 = 0, p5 = 0;
for(int i = 1; i < n; i++)
{
int k2 = dp[p2] * 2;
int k3 = dp[p3] * 3;
int k5 = dp[p5] * 5;
int minm = min(k2, min(k3, k5));
if(minm == k2) p2++;
if(minm == k3) p3++;
if(minm == k5) p5++;
dp.push_back(minm);
}
return dp[n - 1];
}
};