[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)。