给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
思路一:循环遍历,返回第一个大于等于目标值的数的下标,否则返回数组大小
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
for(int i=0;i<=nums.size();++i)
{
if(nums[i]>=target)
return i;
}
return nums.size();
}
};
思路二:运用二分法查找
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
if(nums.size()==0||nums[0]>=target)
return 0;
if(nums[nums.size()-1]<target)
return nums.size();
int beg=0;
int end=nums.size()-1;
int mid=0;
while(beg<=end){
mid=(beg+end)/2;
if(nums[mid]>target)
end=mid-1;
if(nums[mid]<target)
beg=mid+1;
if(nums[mid]==target)
return mid;
}
return nums[mid]>target? mid:mid+1;
}
};