首发于个人博客
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 1:
输入: [1,3,5,6], 5
输出: 2
示例 2:
输入: [1,3,5,6], 2
输出: 1
示例 3:
输入: [1,3,5,6], 7
输出: 4
示例 4:
输入: [1,3,5,6], 0
输出: 0
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:此题比较简单,初读题目感觉是个查找算法。但是有一个非常明确的区别就是可能位于两个数之间,这个就是需要额外注意补充的逻辑。
下面用二分查找来实现一下思路。
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
return searchInsertReal(nums,0,nums.size() - 1,target);
}
int searchInsertReal(vector<int>& nums,int startIndex,int endIndex,int target){
unsigned long size = endIndex - startIndex + 1;
if (size == 0){
return -1;
}
int centerIndex = size / 2 + startIndex;
int centerTarget = nums[centerIndex];
if (centerTarget == target){
//找到了,直接返回结果
return centerIndex;
} else if(centerTarget > target){
if (size > 1){//22222222222
//目标比中间位子数值小,比较前一个,验证是不是在比前一个小。
int preValue = nums[centerIndex - 1];
if (preValue < target){
//小于前一个值,则返回结果
return centerIndex;
}else{
//继续查找,因为已经验证了比前一个还小,则位子可以减1。
return searchInsertReal(nums,startIndex,centerIndex - 1,target);
}
} else{
return centerIndex;
}
} else{
//centerTarget < target
if (size > 2){ //11111111
//目标比中间位子数值大,比较后一个,验证是不是比后一个大
int later = nums[centerIndex + 1];
if (later > target){
//比后一个小,返回结果
return centerIndex + 1;
} else{
//继续查找
return searchInsertReal(nums,centerIndex + 1,endIndex,target);
}
} else{
return centerIndex + 1;
}
}
}
};
上面代码 111111处注释,是size > 2 , 222222处注释是size > 1.
这是因为我们 cneterIndex是 size / 2.如果是size是2的话,centerIndex =1已经是最后一个元素了。所以size最少是3才会有后一个,最少是2才会有前一个。