leetcode丑数（堆加hashset去重）

最新推荐文章于 2022-04-18 11:42:34 发布

小Charles不想菜

最新推荐文章于 2022-04-18 11:42:34 发布

阅读量130

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/weixin_43965055/article/details/119569900

版权

算法专栏收录该内容

5 篇文章 0 订阅

订阅专栏

给你一个整数 n ，请你找出并返回第 n 个丑数。

1.什么是丑数：只包含质因子2，3和5的数

解法：
想要找出第n个丑数，用堆heap可以取出
初始时堆为空。首先将最小的丑数 11 加入堆。

每次取出堆顶元素 xx，则 xx 是堆中最小的丑数，由于 2x, 3x, 5x2x,3x,5x 也是丑数，因此将 2x, 3x, 5x2x,3x,5x 加入堆。

想要不重复，用hashset判断后再存

  public int nthUglyNumber2(int n) {
        int[] yinzi = {2, 3, 5};
        PriorityQueue<Long> heap = new PriorityQueue<>();
        Set<Long> hashset = new HashSet<>();
        hashset.add(1L);
        heap.offer(1L);
        int res = 0;
        for (int i = 0; i < n; i++) {
            long curr = heap.poll();
            res = (int) curr;
            for (long factor : yinzi) {
                long next = curr * factor;
                if (hashset.add(next)) {
                    heap.offer(next);
                }

            }
        }

        return res;
    }

在这里插入图片描述

复盘：速度太慢，是因为堆的重新分布和hashset的条件判断
空间大很正常，java嘛，加上还用了两个数据结构

解法二：动态规划（附上大佬C++代码）

class Solution {
public:
    int nthUglyNumber(int n) {
        vector<int> dp(n);
        dp.at(0)=1;
        int p_2,p_3,p_5;
        p_2=p_3=p_5=0;
        for (int i=1;i<n;++i)
        {
            dp.at(i)=min(min(2*dp.at(p_2),3*dp.at(p_3)),5*dp.at(p_5));
            if (dp.at(i)==2*dp.at(p_2))
                ++p_2;
            if (dp.at(i)==3*dp.at(p_3))
                ++p_3;
            if (dp.at(i)==5*dp.at(p_5))
                ++p_5;
        }
        return dp.at(n-1);
    }
};