LeetCode 264. 丑数 II(优先队列/指针+dp)

2024-02-27 14:04

本文主要是介绍LeetCode 264. 丑数 II(优先队列/指针+dp),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

目录

1.题目

 2.代码

2.1 优先队列

2.2 指针+dp


1.题目

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

丑数 就是只包含质因数 23 和/或 5 的正整数。

示例 1:

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

示例 2:

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

提示:

  • 1 <= n <= 1690

 2.代码

2.1 优先队列

class Solution:def nthUglyNumber(self, n: int) -> int:#3个基础的因数,任何丑数都是由其组成(除了1)arr = [2, 3, 5]#记录出现的数,这个set用的很讲究mp = {1}#优先队列pq = [1]#每个数分别乘以3个因数,避免重复后,再加入优先队列for i in range(n - 1):tmp = heapq.heappop(pq)for k in arr:if (num := tmp * k) not in mp:mp.add(num)heapq.heappush(pq, num) return heapq.heappop(pq)

上面这部分代码其实参考的是官方题解,但是我自己在对其进行改动的时候却发现了一些有意思的事情:

1.当我将mp的数据类型由set改为list的时候,原本170+ms的运行时间变为了1800+ms,经过查阅资料发现:

当执行插入判断操作时set要远比list效率高,因为,虽然两者的时间复杂度都为O(1),但是list在插入数据的过程中,可能会进行重新分配和复制,消耗大量时间。

当执行not in判断操作时set要远比list效率高,因为,哈希表实现的set实现此操作的时间复杂度为O(1),而list则需要O(n)

2.当我将arr的数据类型从list换位set时,执行效率就下降了一些,原因是,进行for in操作时,list的内存是连续的,而set的内存不是连续的,前者对CPU更友好

2.2 指针+dp

class Solution:def nthUglyNumber(self, n: int) -> int:#dp[i]表示第i个丑数是多少dp = [1] * (n + 1)#3个指针dp2 = dp3 = dp5 = 1for i in range(2, n + 1):#选取最小的加入dptmp = min(dp[dp2] * 2, dp[dp3] * 3, dp[dp5] * 5)if tmp == dp[dp2] * 2: dp2 += 1if tmp == dp[dp3] * 3:dp3 += 1if tmp == dp[dp5] * 5:dp5 += 1dp[i] = tmpreturn dp[n]

 老大哥动态规划这场居然没有在执行速度上超过优先队列,震惊。

这篇关于LeetCode 264. 丑数 II(优先队列/指针+dp)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/752642

相关文章

哈希leetcode-1

目录 1前言 2.例题  2.1两数之和 2.2判断是否互为字符重排 2.3存在重复元素1 2.4存在重复元素2 2.5字母异位词分组 1前言 哈希表主要是适合于快速查找某个元素(O(1)) 当我们要频繁的查找某个元素,第一哈希表O(1),第二,二分O(log n) 一般可以分为语言自带的容器哈希和用数组模拟的简易哈希。 最简单的比如数组模拟字符存储,只要开26个c

hdu1180(广搜+优先队列)

此题要求最少到达目标点T的最短时间,所以我选择了广度优先搜索,并且要用到优先队列。 另外此题注意点较多,比如说可以在某个点停留,我wa了好多两次,就是因为忽略了这一点,然后参考了大神的思想,然后经过反复修改才AC的 这是我的代码 #include<iostream>#include<algorithm>#include<string>#include<stack>#include<

hdu4826(三维DP)

这是一个百度之星的资格赛第四题 题目链接:http://acm.hdu.edu.cn/contests/contest_showproblem.php?pid=1004&cid=500 题意:从左上角的点到右上角的点,每个点只能走一遍,走的方向有三个:向上,向下,向右,求最大值。 咋一看像搜索题,先暴搜,TLE,然后剪枝,还是TLE.然后我就改方法,用DP来做,这题和普通dp相比,多个个向上

hdu1011(背包树形DP)

没有完全理解这题, m个人,攻打一个map,map的入口是1,在攻打某个结点之前要先攻打其他一个结点 dp[i][j]表示m个人攻打以第i个结点为根节点的子树得到的最优解 状态转移dp[i][ j ] = max(dp[i][j], dp[i][k]+dp[t][j-k]),其中t是i结点的子节点 代码如下: #include<iostream>#include<algorithm

hdu4865(概率DP)

题意:已知前一天和今天的天气概率,某天的天气概率和叶子的潮湿程度的概率,n天叶子的湿度,求n天最有可能的天气情况。 思路:概率DP,dp[i][j]表示第i天天气为j的概率,状态转移如下:dp[i][j] = max(dp[i][j, dp[i-1][k]*table2[k][j]*table1[j][col] )  代码如下: #include <stdio.h>#include

usaco 1.1 Broken Necklace(DP)

直接上代码 接触的第一道dp ps.大概的思路就是 先从左往右用一个数组在每个点记下蓝或黑的个数 再从右到左算一遍 最后取出最大的即可 核心语句在于: 如果 str[i] = 'r'  ,   rl[i]=rl[i-1]+1, bl[i]=0 如果 str[i] = 'b' ,  bl[i]=bl[i-1]+1, rl[i]=0 如果 str[i] = 'w',  bl[i]=b

uva 10154 DP 叠乌龟

题意: 给你几只乌龟,每只乌龟有自身的重量和力量。 每只乌龟的力量可以承受自身体重和在其上的几只乌龟的体重和内。 问最多能叠放几只乌龟。 解析: 先将乌龟按力量从小到大排列。 然后dp的时候从前往后叠,状态转移方程: dp[i][j] = dp[i - 1][j];if (dp[i - 1][j - 1] != inf && dp[i - 1][j - 1] <= t[i]

uva 10118 dP

题意: 给4列篮子,每次从某一列开始无放回拿蜡烛放入篮子里,并且篮子最多只能放5支蜡烛,数字代表蜡烛的颜色。 当拿出当前颜色的蜡烛在篮子里存在时,猪脚可以把蜡烛带回家。 问最多拿多少只蜡烛。 代码: #include <iostream>#include <cstdio>#include <cstdlib>#include <algorithm>#include <cs

uva 10069 DP + 大数加法

代码: #include <iostream>#include <cstdio>#include <cstdlib>#include <algorithm>#include <cstring>#include <cmath>#include <stack>#include <vector>#include <queue>#include <map>#include <cl

uva 10029 HASH + DP

题意: 给一个字典,里面有好多单词。单词可以由增加、删除、变换,变成另一个单词,问能变换的最长单词长度。 解析: HASH+dp 代码: #include <iostream>#include <cstdio>#include <cstdlib>#include <algorithm>#include <cstring>#include <cmath>#inc