LintCode #4. 丑数 II

最新推荐文章于 2020-09-20 11:01:47 发布

朋至远方

最新推荐文章于 2020-09-20 11:01:47 发布

阅读量189

点赞数

CC 4.0 BY-SA版权

分类专栏：学习笔记文章标签： LintCode

本文链接：https://blog.youkuaiyun.com/qq_33087505/article/details/83758119

学习笔记专栏收录该内容

8 篇文章

订阅专栏

本文提供LintCode上丑数II问题的高效解法，采用堆排序算法，利用heapq模块实现。详解堆排序原理及python堆操作，通过两个列表巧妙维护丑数序列。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

LintCode #4. 丑数 II

[问题地址]：https://www.lintcode.com/problem/ugly-number-ii/description

解法一：

class Solution:
    """
    @param n: An integer
    @return: the nth prime number as description.
    """
    def nthUglyNumber(self, n):
        # write your code here
        import heapq
        hea = [1]
        sex = [2,3,5]
        for i in range(n):
            x = heapq.heappop(hea)
            for j in range(3):
                if x*sex[j] not in hea:
                    heapq.heappush(hea,x*sex[j])
        return x

说明：堆排序的相关知识，python中已经实现在heapq.py中，可以很方便地维护一个堆。优点在于时刻有序，先进先出。其中，hea[]与sex[]，两个列表的设置很巧妙。