leetcode264. 丑数 II(队列应用)

最新推荐文章于 2024-10-20 10:43:07 发布

豆约翰

最新推荐文章于 2024-10-20 10:43:07 发布

阅读量494

点赞数 10

分类专栏：数据结构及算法文章标签：算法 java 数据结构

本文链接：https://blog.csdn.net/douyh/article/details/137234927

版权

数据结构及算法专栏收录该内容

95 篇文章 0 订阅

订阅专栏

丑数 II

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

丑数就是质因子只包含 2、3 和 5 的正整数。

示例 1：

输入：n = 10
输出：12
解释：[1, 2, 3, 4, 5, 6, 8, 9, 10, 12] 是由前 10 个丑数组成的序列。
示例 2：

输入：n = 1
输出：1
解释：1 通常被视为丑数。

提示：

1 <= n <= 1690

class Solution {
    public int nthUglyNumber(int n) {
		if(n==1)
        {
            return 1;
        }

        LinkedList<Integer> twoQueue = new LinkedList<>();
        LinkedList<Integer> threeQueue = new LinkedList<>();
        LinkedList<Integer> fiveQueue = new LinkedList<>();

        List<Integer> lst = new ArrayList<>();
        lst.add(1);
        twoQueue.add(2);
        threeQueue.add(3);
        fiveQueue.add(5);
        n--;
        while (n > 0){
            int a1 = twoQueue.get(0);
            int a2 = threeQueue.get(0);
            int a3 = fiveQueue.get(0);
            int tmp = Math.min(a3,Math.min(a1,a2));
            lst.add(tmp);
            twoQueue.add(tmp*2);
            threeQueue.add(tmp*3);
            fiveQueue.add(tmp*5);
            n--;
            if(tmp == a1){
                twoQueue.removeFirst();
            }
            if(tmp == a2){
                threeQueue.removeFirst();
            }
            if(tmp == a3){
                fiveQueue.removeFirst();
            }
        }
        return lst.get(lst.size()-1);
    }
}

思路

通俗易懂的解释：
首先从丑数的定义我们知道，一个丑数的因子只有2,3,5，那么丑数p = 2 ^ x * 3 ^ y * 5 ^ z，换句话说一个丑数一定由另一个丑数乘以2或者乘以3或者乘以5得到，那么我们从1开始乘以2,3,5，就得到2,3,5三个丑数，在从这三个丑数出发乘以2,3,5就得到4，6,10,6，9,15,10,15,25九个丑数，我们发现这种方法会得到重复的丑数，而且我们题目要求第N个丑数，这样的方法得到的丑数也是无序的。那么我们可以维护三个队列：
（1）丑数数组： 1
乘以2的队列：2
乘以3的队列：3
乘以5的队列：5
选择三个队列头最小的数2加入丑数数组，同时将该最小的数乘以2,3,5放入三个队列；
（2）丑数数组：1,2
乘以2的队列：4
乘以3的队列：3，6
乘以5的队列：5，10
选择三个队列头最小的数3加入丑数数组，同时将该最小的数乘以2,3,5放入三个队列；
（3）丑数数组：1,2,3
乘以2的队列：4,6
乘以3的队列：6,9
乘以5的队列：5,10,15
选择三个队列头里最小的数4加入丑数数组，同时将该最小的数乘以2,3,5放入三个队列；
（4）丑数数组：1,2,3,4
乘以2的队列：6，8
乘以3的队列：6,9,12
乘以5的队列：5,10,15,20
选择三个队列头里最小的数5加入丑数数组，同时将该最小的数乘以2,3,5放入三个队列；
（5）丑数数组：1,2,3,4,5
乘以2的队列：6,8,10，
乘以3的队列：6,9,12,15
乘以5的队列：10,15,20,25
选择三个队列头里最小的数6加入丑数数组，但我们发现，有两个队列头都为6，所以我们弹出两个队列头，同时将12,18,30放入三个队列；