剑指-面试题49 丑数

丑数

题目
我们把只包含因子 2、3 和 5 的数称作丑数（Ugly Number）。求按从小到大的顺序的第 n 个丑数。
1 是丑数。
n 不超过1690

思路 以空间换时间

（1）开辟一个数组存放前面已经排好序的丑数
（2）设置三个位置变量p2、p3、p5表示上一次乘以2、3、5的位置，比如在p2之前的数乘以2都小于当前数组中的最大值，在这个数之后乘以2都大于最大值。
（3）每次新的丑数都是p2对应的值×2、p3对应值×3、p5对应值×5三者中最小的，最后取的哪一个就更新哪个下标。

---

C++

class Solution {
public:
    int nthUglyNumber(int n) {
        if(n<7)
            return n;
        vector<int> res(n);
        res[0]=1;
        int t2=0, t3=0, t5=0;
        for (int i=1; i<n; ++i)
        {
            res[i] = min({res[t2]*2, res[t3]*3, res[t5]*5});  // C11之后
            if (res[i]==res[t2]*2)
                ++t2;
            if (res[i]==res[t3]*3)
                ++t3;
            if (res[i]==res[t5]*5)
                ++t5;
        }
        return res[n-1];
    }
};

python

class Solution:
    def nthUglyNumber(self, n: int) -> int:
        if n<7:
            return n
        res = [1]*n
        p2 , p3, p5 = 0, 0, 0
        for i in range(1, n):
            res[i]=min(res[p2]*2, min(res[p3]*3, res[p5]*5))
            if res[i]==res[p2]*2:
                p2 += 1
            if res[i]==res[p3]*3:
                p3 += 1            
            if res[i]==res[p5]*5:
                p5 += 1         
        return res[n-1]