Search Insert Position 搜索插入位置

给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。

你可以假设数组中无重复元素。

示例 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);
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值