给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n)
的算法。
示例 1:
输入: nums = [1,3,5,6], target = 5
输出: 2
示例 2:
输入: nums = [1,3,5,6], target = 2
输出: 1
示例 3:
输入: nums = [1,3,5,6], target = 7
输出: 4
思路
这道题应该用二分法 但是插入位置要慎重考虑
插入位置有三种情况:对于插到数组前后的两种特殊情况应该单独分类讨论
代码
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
if(target < nums[0]) return 0;//对于插入首尾的两种特殊情况要单独考虑
if(target > nums[nums.size() - 1]) return nums.size();
int left = 0;
int right = nums.size() - 1;
int mid = 0;
while(left <= right){
mid = (left + right) / 2;
if(nums[mid] == target){
return mid;
}else if(nums[mid] < target){
left = mid + 1;
}else{
right = mid - 1;
}
}
return left;
}
};