忙过一阵了,之后全力以赴
今日题目:
1、查找排序矩阵中第k小的数; tag:堆|二分查找
今日摘录:
有一天,蓦然回首,你会发现,那个给你许多痛苦的人,却是你的救赎。
——《借东西的小人阿莉埃蒂》
378. Kth Smallest Element in a Sorted Matrix | Difficulty: Medium
Given a n x n matrix where each of the rows and columns are sorted in ascending order, find the kth smallest element in the matrix.
Note that it is the kth smallest element in the sorted order, not the kth distinct element.
Example:
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
return 13.
tag:堆|二分查找
题意:找到一个矩阵中第k小的数
思路:
1、二分。之前总想着如何去查找,后面一想,无论矩阵如何,只需要对着具体的数去做二分不就行了吗。
class Solution {
public:
int kthSmallest(vector<vector<int>>& matrix, int k) {
int n = matrix.size();
int left = matrix[0][0];
int right = matrix[n-1][n-1];
while(left<right)
{
int mid = left+(right-left)/2;
int count = 0;
for(int i=0;i<n;i++)
{
vector<int>&nums = matrix[i];
count+= upper_bound(nums.begin(),nums.end(),mid)-nums.begin();
}
if(count<k)
{
left = mid+1;
}
else right = mid;
}
return left;
}
};
结果:76ms