数组与矩阵---打印N个数组整体最大的TopK

给定N个有序数组,需要找出这N个数组合并后的前K个最大数。当所有数组元素总数小于K时,打印所有元素。通过构建大根堆,每次取出堆顶元素并更新堆,达到O(KlogN)的时间复杂度。文章介绍了具体的解题思路和代码实现。
摘要由CSDN通过智能技术生成

【题目】

  有N个长度不一的数组,所有的数组都是有序的,请从大到小打印这个N个数组整体最大的前K个数。
  例如,输入含有N行元素的二维数组可以代表N个一维数组。
  219, 405, 538, 845, 971
  148, 558
  52, 99, 348, 691
  再输入整数k = 5,则打印:
  971, 845, 691, 558, 538

【要求】

  1. 如果所有数组的元素个数小于K,则从大到小打印所有的数
  2. 要求时间复杂度为O(KlogN)

【基本思路】

  本题的解法是利用堆结构和堆排序的过程完成的,具体过程如下:

  1. 构建一个大小为N的大根堆,建堆的过程就是把每一个数组的最后一个值,也就是该数组的最大值,依次加入到堆里,这个过程就是建堆的调整过程。

  2. 建好堆以后,此时堆顶的元素就是所有数组中最大的元素,打印堆顶元素。

  3. 假设堆顶的元素来自 a 数组的 i 位置,那么将堆顶的元素用a[i-1]替换,然后从堆的头部重新调整堆。如果发现此时 a 数组已经没有元素,那么就将堆顶元素与堆尾元素交换,同时令堆的大小减1,仍然是从堆的头部重新调整堆。

  4. 每次都可得到一个堆顶元素,打印k个堆顶元素,就是最终的结果。

为了知道每一次的堆顶元素来自哪一个数组的哪一个位置,我们可以定义如下的heap类:

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值