剑指 Offer 49. 丑数

最新推荐文章于 2022-04-09 20:15:44 发布

还记得樱花正开~

最新推荐文章于 2022-04-09 20:15:44 发布

阅读量360

点赞数

分类专栏： leetcode 文章标签： leetcode 动态规划算法

本文链接：https://blog.csdn.net/ZHUYAN1209/article/details/123476251

版权

leetcode 专栏收录该内容

223 篇文章 2 订阅

订阅专栏

本文介绍了两种解决LeetCode中求解第n个丑数问题的方法，分别是优先队列（小根堆）和动态规划。优先队列的方法通过不断存储和更新2、3、5的倍数，避免重复并确保找到第n个丑数。动态规划则利用三个指针跟踪2、3、5的乘积，存储最小的下一个丑数。两种方法都有效地解决了问题。

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

题目

力扣

思路优先队列/小根堆

先存后排，先把ans置为1，每次分别把ans的2、3、5倍存入优先队列，取出次数为n时，ans就是答案。要注意去重，每次循环的最后，把结果与ans相等的数据取出来。还要注意数据越界，要用long来存。

代码

class Solution {
public:
    int nthUglyNumber(int n) {
        priority_queue<long, vector<long>, greater<long>> pq;
        long ans = 1;
        for(int i = 1; i < n ; i++){
            pq.push(ans * 2);
            pq.push(ans * 3);
            pq.push(ans * 5);
            ans = pq.top();
            pq.pop();
            while(!pq.empty() && pq.top() == ans)
                pq.pop();
        }
        return ans;
    }
};

思路动态规划

先排后存，用三指针表示2、3、5乘这个数得到的结果正准备添加到dp数组中。

代码

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