二分法搜索结束定位
查找边界
1.搜数组起始位置
class Solution {
public:
int binarySearch(vector<int>& nums, int target, bool lower) {
int left = 0, right = (int)nums.size() - 1, ans = (int)nums.size();
while (left <= right) {
int mid = (left + right) / 2;
if (nums[mid] > target || (lower && nums[mid] >= target)) {
right = mid - 1;
ans = mid;
} else {
left = mid + 1;
}
}
return ans;
}
vector<int> searchRange(vector<int>& nums, int target) {
int leftIdx = binarySearch(nums, target, true);
int rightIdx = binarySearch(nums, target, false) - 1;
if (leftIdx <= rightIdx && rightIdx < nums.size() && nums[leftIdx] == target && nums[rightIdx] == target) {
return vector<int>{leftIdx, rightIdx};
}
return vector<int>{-1, -1};
}
};
1.结束位置left=right+1;
2.如果数存在:如果为false(寻找上边界),结束位置,结束位置right一定等于target;相反(寻找下边界),结束位置left一定等于target;
3.如果数不存在,如果为false(寻找上边界),结束位置根据target,target为最大值则则left一定出上边界,target为最小值则则right一定出下边界。
2.搜二维数组
class Solution {
public:
bool searchMatrix(vector<vector<int>> matrix, int target) {
auto row = upper_bound(matrix.begin(), matrix.end(), target, [](const int b, const vector<int> &a) {
return b < a[0];
});
if (row == matrix.begin()) {
return false;
}
--row;
return binary_search(row->begin(), row->end(), target);
}
};
#include
binary_search:查找某个元素是否出现。
binary_search(arr[],arr[]+size , indx)
lower_bound:查找第一个大于或等于某个元素的位置。
lower_bound(arr[],arr[]+size , indx)
upper_bound:查找第一个大于某个元素的位置。
lower_bound(arr[],arr[]+size , indx)
[](const int b, const vector &a) {
return b < a[0];
}
lambda公式 特定格式,返回值auto自动补全
二分法模板
正常这种直接三目运算符这种格式判断结束位置
int searchInsert(vector<int>& nums, int target) {
int l=0,r=nums.size()-1;
int mid=(l+r)/2;
while(l<r){
if(nums[mid]<target) l=mid+1;
else if(nums[mid]>target) r=mid;
else return mid;
mid=(l+r)/2;
}
return nums[l] < target? l+1 : l;
}