一.题目描述
给定一个排序数组nums(无重复元素)与目标值target,如果target在nums里出现,则返回target所在下标,如果target在nums里面没有出现,则返回target应该插入位置的数组下标,使得target插入数组nums之后,数组任然有序。
二.解题思路
只是在二分查找的基础上添加了几个判断条件,以有序序列“1 3 5 6”为例来说明:
如果目标值为小于nums[mid]=3, 但是大于nums[mid-1]=1, 例如当目标值为2时,则返回的位置为mid;
如果目标值大于nums[mid]=3,但是小于nums[mid+1]=5, 例如当目标值为4时,则返回的位置为mid+1;
然后还要考虑边界,即当mid=0时,说明目标值比nums[0]还小,则应该返回mid;当mid=nums.size()-1时说明目标值比nums[mid]还要大,则返回mid+1;
三.实现代码
/*
*功能:返回目标值应该插入的位置
*时间:2019.7.31---11:09
*作者:Beson-NHW
*
*/
#include<iostream>
#include<vector>
using namespace std;
class Solution{
public:
int searchInsert(vector<int>&nums,int target){
int index=-1;
int begin=0;
int end=nums.size()-1;
while(index==-1){
int mid=(begin+end)/2;
if(target==nums[mid]){
index=mid;
}
else if(target>nums[mid]){
if(mid==nums.size()-1||target<nums[mid+1]){
index=mid+1;
}
begin=mid+1;
}
else if(target<nums[mid]){
if(mid==0||target>nums[mid-1]){
index=mid;
}
end=mid-1;
}
}
return index;
}
};
int main()
{
int b[10]={12,23,56,69,78,90,100,123,132,156};
vector<int>a(b,b+10);
Solution solve;
cout<<solve.searchInsert(a,157)<<endl;
return 0;
}