天天看点

动态规划练习 leetcode 264.丑数Ⅱ

leetcode 264.丑数Ⅱ 动态规划

动态规划练习 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];
        
    }
};