给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
思路:
由于给定的是排序数组,所以最笨的办法是顺序查找,当找到大于或等于target值时,或者到数组末尾依旧不满足条件,返回下标(到数组末尾返回nums,size()即可)。但是这样的时间效率是O(n),给定的是排序数组我们可以用二分查找把时间复杂度降到O(logn)。
由于题目给了几个样例并且不是查找target是否在数组中,二是查找第一个大于等于target的下标,所以传统的二分查找要修改一下,主要是边界条件的修改,我们传入判断递归的函数包括这样几个参数:
nums //数组
start //递归开始下标
end //递归结束下标
边界条件判断如下:
1:如果nums[start]>=target,则返回start
2:如果nums[end]<target,则返回end+1
3:如果nums[end]=target,则返回end
4:判断mid,分三种情况:如果nums[mid]==target,则返回mid
5:如果nums[mid]<target,则递归调用[mid+1,end]的子函数
6:如果nums[mid]>target,则递归调用[start,mid-1]的子函数
代码如下:
int searchInsertCore(vector<int>& nums, int target, int start, int end) {
if (nums[start] >= target) {
return start;
}
if (nums[end] < target) {
return end + 1;
}
if (nums[end] == target) {
return end;
}
int mid = (start + end) / 2;
if (nums[mid] == target) {
return mid;
}
if (nums[mid] > target) {
if ((mid - 1) >= start) {
return searchInsertCore(nums, target, start, mid - 1);
}
}
if (nums[mid] < target) {
if ((mid + 1) <= end) {
return searchInsertCore(nums, target, mid+1, end);
}
}
return -1;
}
int searchInsert(vector<int>& nums, int target) {
if (nums.size() <= 0) {
return 0;
}
return searchInsertCore(nums,target,0,nums.size()-1);
}