剑指offer 50. 丑数

最新推荐文章于 2024-07-25 11:27:31 发布

Pandaconda

最新推荐文章于 2024-07-25 11:27:31 发布

阅读量281

点赞数

分类专栏： # 剑指offer第二版文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/Newin2020/article/details/126997872

版权

剑指offer第二版专栏收录该内容

77 篇文章 8 订阅

订阅专栏

✍个人博客：https://blog.csdn.net/Newin2020?spm=1011.2415.3001.5343
📚专栏地址：剑指offer系列题解
📝原题地址：题目地址
📣专栏定位：为找工作的小伙伴整理常考算法题解，祝大家都能成功上岸！
❤️如果有收获的话，欢迎点赞👍收藏📁，您的支持就是我创作的最大动力💪

题目描述

我们把只包含质因子 2、3 和 5 的数称作丑数（Ugly Number）。

例如 6、8 都是丑数，但 14 不是，因为它包含质因子 7。

求第 n 个丑数的值。

数据范围

1≤n≤1000

样例
输入：5

输出：5
注意：习惯上我们把 1 当做第一个丑数。

方法一：动态规划 O(n)

我们可以利用多路归并思想，动态的更新值。

假设一个序列 S ，其中包含了所有丑数即只包含质因子 2 、3 和 5 的数。那么如果那么现在有三个序列，这三个序列分别只包含质因子 2 、3 和 5 的数，则这三个序列并集就等于序列 S 。所以我们可以用三个指针分别指向这三个序列的第一个元素，每次都只取这三个指针指向的最小值加入答案数组，并将该指针往后移，直到得到第 n 个丑数。

我们可以对上述思路进行优化，只用一个数组来存储得到的丑数，三个指针指向开头，每得到一个最小值就加入该数组末尾，然后将指针后移一位再进行比较，这样可以保证所有数都不会被忽略。我们拿题目的样例 n = 5 举例，来看看具体的思路是如何：

第一步： 先将 1 加入数组中，因为这是一个特殊的样例，需要进行初始化，且将 i 、j 和 k 都指向 q[0] 。其中 i 用于指向只包含质因数 2 的序列，j 则指向只包含质因数 3 的序列，k 则指向只包含质因数 5 的序列。

在这里插入图片描述

第二步： 从三个指针指向的元素中取最小值，即 $t = min (1 * 2, 1 * 3, 1 * 5) = min (2, 3, 5) = 2$ ，故将 i 往后移一位。

在这里插入图片描述

第三步： 从三个指针指向的元素中取最小值，即 $t = min (2 * 2, 1 * 3, 1 * 5) = min (4, 3, 5) = 3$ ，故将 j 往后移一位。

在这里插入图片描述

第四步： 从三个指针指向的元素中取最小值，即 $t = min (2 * 2, 2 * 3, 1 * 5) = min (4, 6, 5) = 4$ ，故将 i 往后移一位。

在这里插入图片描述

第五步： 从三个指针指向的元素中取最小值，即 $t = min (3 * 2, 1 * 3, 1 * 5) = min (6, 6, 5) = 5$ ，故将 k 往后移一位。

在这里插入图片描述
第六步: 此时容器中已经有 5 个元素，直接返回最后一个元素 5 即可。

class Solution {
public:
    int getUglyNumber(int n) {
        vector<int> q(1, 1); //初始化第一个丑数1
        int i = 0, j = 0, k = 0;
        while (--n)
        {
            int t = min(q[i] * 2, min(q[j] * 3, q[k] * 5));
            q.push_back(t);
            if (q[i] * 2 == t) i++;
            if (q[j] * 3 == t) j++;
            if (q[k] * 5 == t) k++;
        }
        return q.back();
    }
};