给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
解法:
(1)遍历搜索法
这种方法简单粗暴,直接进行全局遍历寻找目标元素
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int i;
for(i = 0;i < nums.size();i++)
{
if(nums[i] >= target)
return i;
}
return i;
}
};
(2)二分查找法
对于题目中这种有序序列,当然可以使用经典的二分查找算法
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
if(nums.size() == 0)return 0;
int left = 0,right = nums.size() - 1,mid;
while(left <= right)
{
//mid = (left + right)/2 ;
mid = left + (right-left)/2;
if(nums[mid] == target) return mid;
else if(nums[mid] < target) left = mid+1;
else right = mid -1;
}
return left;
}
};
需要注意的是,这里最后只能返回left,返回right会报错。