赞
踩
题目描述:
给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第k小的元素。
请注意,它是排序后的第k小元素,而不是第k个元素。
示例:
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。
说明:
你可以假设 k 的值永远是有效的, 1 ≤ k ≤ n2 。
代码:
class Solution {
public:
int kthSmallest(vector<vector<int>>& matrix, int k) {
priority_queue<int,vector<int>,less<int> >q;
for(int i=0;i<matrix.size();i++){
for(int j=0;j<matrix[i].size();j++){
q.push(matrix[i][j]);
if(q.size()>8)
q.pop();
}
}
return q.top();
}
};
使用的是优先队列。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。