2022-7-9 剑指offer-优先队列-hashmap

剑指 Offer 49. 丑数

难度中等360收藏分享切换为英文接收动态反馈

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

class Solution {
    public int nthUglyNumber(int n) {
        PriorityQueue<Integer> queue=new PriorityQueue<>();
        Set<Integer> set=new HashSet<>();
        if (n==1) return 1;
        queue.offer(1);
        set.add(1);
        for (int i=0;i<n-1;i++){
            int temp=queue.poll();
            if (temp<Integer.MAX_VALUE/2&&!set.contains(temp*2)) {
                queue.offer(temp*2);
                set.add(temp*2);
            }
            if (temp<Integer.MAX_VALUE/3&&!set.contains(temp*3)) {
                queue.offer(temp*3);
                set.add(temp*3);
            }
            if (temp<Integer.MAX_VALUE/5&&!set.contains(temp*5)) {
                queue.offer(temp*5);
                set.add(temp*5);
            }

        }
        return queue.poll();
    }
}

思路：丑数总是由丑数*2 *3 *5诞生的，可以用优先队列存储丑数，构造新的丑数，主要要用hashmap去重和int溢出