一.问题描述
Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.
You may assume no duplicates in the array.
Here are few examples.
[1,3,5,6], 5 → 2
[1,3,5,6], 2 → 1
[1,3,5,6], 7 → 4
[1,3,5,6], 0 → 0
二.我的解题思路
拿到这个题目,自然是可以延续上一道题目的思路,使用二分法来尽快找到target。相比于上一题,这道题比较简单。测试通过的程序如下:
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int len=nums.size();
int idx=-1;
int left=0;int right=len-1;int mid=0;int left_idx=-1;int right_idx=len;
if(target<nums[0]) return 0;
if(target>nums[len-1]) return len;
mid=(left+right)/2;
//首先使用二分法寻找数组nums中可有target,并且记录左右边界:nums[left_idx]<target和nums[right_idx]>target
while(nums[mid]!=target)
{
if(nums[mid]>target)
{right=mid;right_idx=right<right_idx?right:right_idx;}
if(nums[mid]<target)
{left=mid;left_idx=left>left_idx?left:left_idx;}
mid=(left+right)/2;
if(right-left==1){
if(nums[left]==target)
{idx = left;break;}
if(nums[right]==target)
{idx = right;break;}
break;
}
if(left==right)
break;
}
if(nums[mid]==target)
idx=mid;
if(idx==-1)
return right;
return idx;
}
};