题目链接:35. 搜索插入位置 - 力扣(LeetCode)
这里和正常二分的区别就是如果这个数字不是数组中的数字应该把它插在哪,我们二分搜索出来的数要么比目标值大,要么比目标值小,这时候我们要判断一下:
如果搜索出来的数nums[mid]比target大,那就插入在mid位置。
如果搜索出来的数nums[mid]比target小,那就插入在mid+1位置。
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int l=0;
int r=nums.size()-1;
int mid;
while(l<=r){
mid=(l+r+1)/2;
if(nums[mid]==target)
return mid;
else if(nums[mid]<target)
l=mid+1;
else
r=mid-1;
}
if(nums[mid]>target)
return mid;
return mid+1;
}
int main(){
int x;
vector<int> nums;
int tar;
while (cin>>x)
{
nums.push_back(x);
}
nums.pop_back();
tar = x;
cout << searchInsert(nums, tar);
return 0;
}
};
然后优化一下,相当于每次位置都是r+1
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int l=0;
int r=nums.size()-1;
int mid;
while(l<=r){
mid=(l+r+1)/2;
if(nums[mid]<target)
l=mid+1;
else
r=mid-1;
}
return r+1;
}
int main(){
int x;
vector<int> nums;
int tar;
while (cin>>x)
{
nums.push_back(x);
}
nums.pop_back();
tar = x;
cout << searchInsert(nums, tar);
return 0;
}
};
题目链接:34. 在排序数组中查找元素的第一个和最后一个位置 - 力扣(LeetCode)
这道题他的答案类型我们先列举出来
(1)[2] 2
(2)[1,2,4,4,6] 3
(3)[1,2,3,4,5,5,6] 5
(4)[] 0
也就这几种情况需要考虑
首先最先考虑的就是情况(3),searchRanger函数是找最右区域的,searchRangel函数是找最左区域的,这个就是根据之前的二分题目可以很简单推出来,他的结果也就是{a,b}
其次考虑的是情况(2)这种情况,他的左边界会比右边界大,因为这连个函数在l=r的下一步,r会往左移一步所以造成左边界会比右边界大,所以a>b 答案是-1 -1
其次考虑情况(1),它的左边界会出界,因为它不会走lans=l,所以它还是会是初值-1,右界值对的
最后情况(4),由于未赋值,数组值是0,导致右边界值不会走rans=r,所以它还是会是初值-1,它也是a>b所以不用多加一个判断了得{-1,-1}
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
int searchRanger(vector<int>& nums, int target) {
int l = 0;
int rans = -1;
int r = nums.size() - 1;
int mid;
while (l <= r) {
mid = l + (r - l) / 2;
if (nums[mid] <= target) {
l = mid + 1;
} else
r = mid - 1;
rans = r;
}
return rans;
}
int searchRangel(vector<int>& nums, int target) {
int l = 0;
int lans = -1;
int r = nums.size() - 1;
int mid;
while (l <= r) {
mid = l + (r - l) / 2;
if (nums[mid] < target)
{l = mid + 1;
lans = l;}
else {
r = mid - 1;
}
}
return lans;
}
vector<int> searchRange(vector<int>& nums, int target) {
int a=searchRangel(nums, target);
int b=searchRanger(nums, target);
if(a>b) return{-1,-1};
else if(a<b && a<0) return {0,b};
return {a,b};
}
int main() {
int x;
vector<int> nums;
int tar;
while (cin >> x) {
nums.push_back(x);
}
nums.pop_back();
tar = x;
searchRange(nums, x);
return 0;
}
};