Day28 49. 丑数

最新推荐文章于 2024-10-08 00:02:21 发布

Tracker123

最新推荐文章于 2024-10-08 00:02:21 发布

阅读量227

点赞数

文章标签： c++

本文链接：https://blog.csdn.net/chenshushan3/article/details/124500941

版权

剑指offer 专栏收录该内容

55 篇文章 0 订阅

订阅专栏

本文介绍了如何解决LeetCode上的第49题——丑数问题。提供了两种解题方法，一种是利用最小堆，另一种是采用动态规划。这两种方法分别通过维护质因子2、3和5的倍数，找到第n个丑数。示例中展示了当n为10时，第10个丑数为12。

摘要由CSDN通过智能技术生成

49. 丑数

文章目录

[49. 丑数](https://leetcode-cn.com/problems/chou-shu-lcof/)
- 解题思路

我们把只包含质因子 2、3 和 5 的数称作丑数（Ugly Number）。求按从小到大的顺序的第 n 个丑数。

示例:
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。

解题思路

方法一：最小堆

class Solution {
public:
    int nthUglyNumber(int n) {
        vector<int> factors = {2, 3, 5};
        unordered_set<long> seen;
        priority_queue<long, vector<long>, greater<long>> heap;
        seen.insert(1L);
        heap.push(1L);
        int ugly = 0;
        for(int i = 0; i < n; ++i) {
            long cur = heap.top();
            heap.pop();
            ugly = (int)cur;
            for(int factor : factors) {
                long next = cur * factor;
                if(!seen.count(next)) {
                    seen.insert(next);
                    heap.push(next);
                }
            }
        }
        return ugly;
    }
};

方法二：动态规划

class Solution {
public:
    int nthUglyNumber(int n) {
        vector<int> dp(n+1);
        dp[1] = 1;
        int p2 = 1, p3 = 1, p5 = 1;
        for(int i = 2; i <= n; i++) {
            int num2 = dp[p2] *2, num3 = dp[p3] * 3, num5 = dp[p5] * 5;
            dp[i] = min(min(num2, num3), num5);
            if(dp[i] == num2) p2++;
            if(dp[i] == num3) p3++;
            if(dp[i] == num5) p5++;
        }
        return dp[n];
    }
};