378有序矩阵中第k小的元素解法之优先队列

378 有序矩阵中第k小的元素

给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第 k 小的元素。请注意,它是排序后的第 k 小元素,而不是第 k 个不同的元素。

示例
matrix = [
   [ 1,  5,  9],
   [10, 11, 13],
   [12, 13, 15]
],
k = 8,

返回 13。

归并排序的思路

使用了堆的性质。使用小根堆,已知堆的几种算法的时间复杂度如下:push/O(log(n)), pop/O(log(n)), top/O(log(n))

这里的堆由于使用了树形结构,优化了其pop以及top的时间复杂度(n到logn)。

**解题思路。**先将matrix的第0列传入堆中,然后从中pop出最小的那个,记录下pop的是哪个,然后传入其在矩阵中右边相邻的数。这样进行k次,第k次top出的就是最小的那个。


关于堆的用法如下:

priority_queue<point, vector<point>, greater<point>> que;
// priority_queue优先队列即堆
// 小根堆如上是三个参数,后两个参数不写默认是大根堆
// 小根堆是小的在堆顶,大根堆是大的在堆顶
// 要注意,上面的“point”是一个结构体,所以要在结构体里重载大于号
// 如果是大根堆,需要在结构体里重载小于号。重载大于号形式如下:
bool operator> (const point& a) const { return this->val > a.val; }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值