35.搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
二分法
- 还是用二分法搜索,如有就返回mid。
- 如果没有就和退出循环时的mid比较,确定插入位置。
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
if(nums.size()==0 || target < nums[0]) return 0;
if(nums[nums.size()-1] < target) 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 if(nums[mid] > target) right = mid-1;
}
if(nums[mid] < target) return mid+1;
return mid;
}
};
通过时间: