704. 二分查找
1. 提交时报溢出错误,太久没写C++忘记数组上限应该是n-1。
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0;
int right = nums.size() - 1; // 定义target在左闭右闭的区间里,[left, right]
while (left <= right) { // 当left==right,区间[left, right]依然有效,所以用 <=
int middle = left + ((right - left) / 2);// 防止溢出 等同于(left + right)/2
if (nums[middle] > target) {
right = middle - 1; // target 在左区间,所以[left, middle - 1]
} else if (nums[middle] < target) {
left = middle + 1; // target 在右区间,所以[middle + 1, right]
} else { // nums[middle] == target
return middle; // 数组中找到目标值,直接返回下标
}
}
// 未找到目标值
return -1;
}
};
类似问题:35. 搜索插入位置
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int left = 0;
int right = nums.size() - 1;
while(left <= right){
int middle = left + ((right - left) / 2);
if (nums[middle] < target){
left = middle + 1;
}
else if (nums[middle] > target){
right = middle - 1;
}
else{
return middle;
}
}
return right + 1;
}
};
34. 在排序数组中查找元素的第一个和最后一个位置
这个问题需要拆分来看。直接查找时间复杂度为O(n),不符合题意,故排除。
考虑到时间复杂度问题,可以使用两次二分法分别查找开始位置和结束位置。
第一次二分法查找开始位置,第二次二分法查找结束位置。
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
vector<int> res(2, -1);
res[0] = getLeftBorder(nums, target);
res[1] = getRightBorder(nums, target);
if(res[0] > res[1] || res[0] == -1 || res[1] == -1){
res = {-1,-1};
}
return res;
}
private:
int getRightBorder(vector<int>& nums, int target) {
int left = 0;
int right = nums.size() - 1;
int rightBorder = -1;
while(left <= right){
int middle = left + ((right - left) >> 2);
if(nums[middle] > target){
right = middle - 1;
}
else{
rightBorder = middle;
left = middle + 1;
}
}
return rightBorder;
}
int getLeftBorder(vector<int>& nums, int target) {
int left = 0;
int right = nums.size() - 1;
int leftBorder = -1;
while(left <= right){
int middle = left + ((right - left) >> 2);
if(nums[middle] < target){
left = middle + 1;
}
else{
leftBorder = middle;
right = middle - 1;
}
}
return leftBorder;
}
};