一、题目介绍
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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) {
int n = nums.size();
if(n == 0) //数组为空的情况
return 0;
int l = 0;
int r = n-1;
while(l <= r)
{
int mid = l + (r-l)/2;
if(nums[mid] == target)
return mid;
else if(nums[mid] > target)
r = mid - 1;
else
l = mid + 1;
}
if(l == n) //目标值大于数组中所有元素,此时l = n,防止越界访问
return l;
if(nums[l] > target)
return l;
else
return l+1;
}
};