LeetCode第35题:搜索插入位置
题目描述:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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在数组中,最终会返回它的位置。
如果target不在数组中,此时low==high,如果low所指位置<target,则插入位置应为low+1;
否则low即为target应该插入的位置。
C++代码:
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int low = 0,high = nums.size()-1;
int middle;
//二分查找
while(low<high){
middle = (low+high)>>1;
if(nums[middle]<target)
low = middle+1;
else if(nums[middle]>target)
high = middle-1;
else
return middle;
}
//二分查找未找到target,此时low=high
//若low位置<target,应前进一格
//否则low即为插入位置
if(nums[low]<target)
return low+1;
else
return low;
}
};