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; }