leetcode_313超级丑数--未解决的动态规划方案

题目描述

超级丑数 是一个正整数，并满足其所有质因数都出现在质数数组 primes 中。

给你一个整数 n 和一个整数数组 primes ，返回第 n 个 超级丑数 。

题目数据保证第 n 个 超级丑数 在 32-bit 带符号整数范围内。

示例 1：

输入：n = 12, primes = [2,7,13,19]
输出：32
解释：给定长度为 4 的质数数组
primes =[2,7,13,19]，前 12 个超级丑数序列为：[1,2,4,7,8,13,14,16,19,26,28,32] 。

示例 2：

输入：n = 1, primes = [2,3,5]
输出：1
解释：1 不含质因数，因此它的所有质因数都在质数数组 primes = [2,3,5] 中。

提示：

1 <= n <= 106
1 <= primes.length <= 100
2 <= primes[i] <= 1000
题目数据 保证 primes[i] 是一个 质数
primes 中的所有值都 互不相同 ，且按 递增顺序 排列

来源：力扣（LeetCode）

题目大意

给定一个只包含质数的数组primes和整数n，求第n位的丑数。

丑数：一个数的质因子都在primes数组中叫丑数

思路描述

两种解决方式：

1. 优先队列：小根堆（PriorityQueue）+ 哈希表（HashSet）
2. 动态规划：三指针

优先队列：

1. 建立PriorityQueue和HashSet
2. 将 PriorityQueue和HashSet初始化
3. 找到堆顶元素temp
4. 将给的数组primes中的每个数和堆顶元素相乘
5. 判断相乘后的结果是否在小根堆里已有
6. 如果没有加入到小根堆中
7. 重复3-6操作直到找到第n位丑数
8. 返回丑数

优先队列时间复杂度： O(nmlognm) n为找的第n位丑数，m为给定数组长度

代码

优先队列代码：

class Solution {
	    public int nthSuperUglyNumber(int n, int[] primes) {
	    	
	    	//1. 建立PriorityQueue和HashSet
	    	PriorityQueue<Long> queue=new PriorityQueue<>();
	    	HashSet<Long> set=new HashSet<>();
	    	//2. 将 PriorityQueue和HashSet初始化
	    	queue.add((long) 1);
	    	set.add((long) 1);
	    	
	    	//7. 重复3-6操作直到找到第n位丑数
	    	for(int i=0;i<n-1;i++){
	    	    //3. 找到堆顶元素temp
	    		long temp=queue.poll();
	    		//4. 将给的数组primes中的每个数和堆顶元素相乘
                for(int j:primes){
                //5. 判断相乘后的结果是否在小根堆里已有
                    long temp1=j*temp;
                    if(set.add(temp1)){
                    //6. 如果没有加入到小根堆中
	    			    queue.offer(temp1);
	    		    }
                }
	    		
	    	}
	    	//8. 返回丑数
	    	long i=queue.poll();
	    	return (int)i;

	    }
	    
	    
	}

优先队列运行结果：

写在最后

今日感受

如果感觉写的还不错的话不妨分享给其他人，以求共同进步！

文章如有错误之处请指出，我会听取并改正！

特别说明： 引用部分只用于学习无商业用途