剑指 Offer -- 丑数(三十三)

丑数(三十三)

题目描述

把只包含质因子 2、3 和 5 的数称作丑数（Ugly Number）。例如 6、8 都是丑数，但 14 不是，因为它包含质因子 7。 习惯上我们把 1 当做是第一个丑数。求按从小到大的顺序的第 N 个丑数。

代码(已在牛客上 AC)

使用额外的空间记录第 i 个丑数是多少, 同时引入 p1, p2 和 p3 三个索引, 方便按从小到大的顺序求出丑数. 具体详见代码.

class Solution {
public:
    int GetUglyNumber_Solution(int index) {
        if (index <= 0) return 0;
        vector<int> k(index);
        k[0] = 1;
        int p1 = 0, p2 = 0, p3 = 0;
        for (int i = 1; i < index; ++i) {
            k[i] = min(min(2 * k[p1], 3 * k[p2]), 5 * k[p3]);
            if (k[i] == 2 * k[p1]) p1 ++;
            if (k[i] == 3 * k[p2]) p2 ++;
            if (k[i] == 5 * k[p3]) p3 ++;
        }
        return k[index - 1];
    }
};