LeetCode#378有序矩阵中第k小元素

题目描述

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

示例:

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

返回 13。
说明:
你可以假设 k 的值永远是有效的, 1 ≤ k ≤ n2 。

简单分析

想到最小堆，试了一下过了，但是因为我最小堆好像只能一维数组，这个转换为一维又空间占用很多的样子emmmm有序二维数组这个特点应该可以利用一下

代码

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
    vector<int> nums;
    for(int i=0;i<matrix.size();i++){
        for(int j=0;j<matrix.size();j++){
            nums.push_back(matrix[i][j]);
        }
    }
    make_heap(nums.begin(),nums.end(),greater<int>());
    while(k>1){
        pop_heap(nums.begin(),nums.end(),greater<int>());
        nums.pop_back();
        k--;
    }
    return nums[0];
}
};