返回顶部

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

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());
        return sorted[k - 1];

    }
};

思路

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

posted @ 2020-07-02 12:18  Swetchine  阅读(143)  评论(0编辑  收藏  举报