给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第k小的元素。 请注意,它是排序后的第k小元素,而不是第k个元素。
示例:
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。
说明: 你可以假设 k 的值永远是有效的, 1 ≤ k ≤ n2 。
int kthSmallest(vector<vector<int>>& matrix, int k) {
priority_queue<int,vector<int>,less<int>> result;
int len = matrix[0].size();
int count = 0;
for (unsigned i = 0; i < len; ++i) {
for (unsigned j = 0; j < len; ++j) {
if (result.size() >= k) {
if(result.top()>matrix[i][j]){
result.push(matrix[i][j]);
result.pop();
}
}
else
result.push(matrix[i][j]);
}
}
return result.top();
}执行用时 :48 ms, 在所有 cpp 提交中击败了67.71%的用户
内存消耗 :13.1 MB, 在所有 cpp 提交中击败了34.15%的用户