[leetcode]378. 有序矩阵中第K小的元素

Posted 2020-12-14 斯卡堡

378. 有序矩阵中第K小的元素

Difficulty: 中等

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

示例：

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

返回 13。

提示：
你可以假设 k 的值永远是有效的，1 ≤ k ≤ n<sup>2?</sup>。

解法一：暴力

?class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        int n = matrix.size();
        vector<int> sorted;
        for(int i = 0;i<n;++i){
            for(int j = 0;j<n;++j){
                sorted.push_back(matrix[i][j]);
            }
        }
        sort(sorted.begin(),sorted.end());
        for(int i : sorted){
            cout<< i<< " ";
        }
        return sorted[k - 1];

    }
};

思路

取出数组全部元素再进行排序，再拿出第k小的数即可，时间复杂度主要耗费在遍历二维数组的时间上了