算法:堆(优先队列)

堆(优先队列)分为最大堆和最小堆。

python实现

class HEAPQ:
	# 最小堆
    def __init__(self, l):
        self.l = l
        self.build()

    def build(self):
        n = len(self.l)
        end = n // 2
        for i in range(end, -1, -1):
            self.update(i)

    def update(self, i):
        l = self.l
        n = len(l)
        idx = i
        if 2 * i + 1 < n and l[2 * i + 1] < l[idx]:
            idx = 2 * i + 1
        if 2 * i + 2 < n and l[2 * i + 2] < l[idx]:
            idx = 2 * i + 2
        if idx != i:
            l[i], l[idx] = l[idx], l[i]
            self.update(idx)

    def push(self, x) -> int:
        l = self.l
        l.append(x)
        i = len(l)-1
        while i != 0:
            parent = (i-1) // 2
            # 父节点大于 则上浮
            if l[parent] > l[i]:
                l[parent], l[i] = l[i], l[parent]
                i = parent
            else:
                break

    def pop(self):
        l = self.l
        n = len(l)
        l[0], l[n - 1] = l[n - 1], l[0]
        t = l.pop()
        self.update(0)
        return t

    def top(self):
        return self.l[0]

215. 数组中的第K个最大元素

给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。
请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。

215. 数组中的第K个最大元素

面试题 17.09. 第 k 个数

有些数的素因子只有 3,5,7,请设计一个算法找出第 k 个数。注意,不是必须有这些素因子,而是必须不包含其他的素因子。例如,前几个数按顺序应该是 1,3,5,7,9,15,21。
面试题 17.09. 第 k 个数

class Solution:
    def getKthMagicNumber(self, k: int) -> int:
        l = [1]
        ss = set()
        tt = (3,5,7)
        for i in range(1,k):
            t = heapq.heappop(l)
            for x in tt:
                if t*x not in ss:
                    heapq.heappush(l, t*x)
                    ss.add(t*x)
        return l[0]

相关推荐

  1. 算法优先队列

    2024-04-22 02:46:03       14 阅读
  2. 502. IPO(贪心算法+优先队列/

    2024-04-22 02:46:03       29 阅读
  3. 优先级队列

    2024-04-22 02:46:03       18 阅读

最近更新

  1. TCP协议是安全的吗?

    2024-04-22 02:46:03       16 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-04-22 02:46:03       16 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-04-22 02:46:03       15 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-04-22 02:46:03       18 阅读

热门阅读

  1. Beego框架学习

    2024-04-22 02:46:03       14 阅读
  2. 蓝桥杯刷题-包子凑数

    2024-04-22 02:46:03       14 阅读
  3. 好数(蓝桥杯)

    2024-04-22 02:46:03       13 阅读
  4. python面向对象的使用(2)

    2024-04-22 02:46:03       13 阅读
  5. 嵌入式系统中AI可以提高开发效率方法

    2024-04-22 02:46:03       12 阅读
  6. 服务器为什么要定期备份

    2024-04-22 02:46:03       16 阅读
  7. 在线测试与论文管理系统

    2024-04-22 02:46:03       15 阅读
  8. Linux:RHCE之web服务练习

    2024-04-22 02:46:03       13 阅读
  9. 案例分析常见问题

    2024-04-22 02:46:03       12 阅读