题目描述
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
简单分析
对于有序数组,暴力一点解法就是直接遍历,若遇到目标则返回,若第一次遇到比目标更大的数字而证明没有目标值,且目标应位于当前位置。
代码
//暴力法
int searchInsert(vector<int>& nums, int target) {
if(nums.size()==0)
return 0;
for(int i=0;i<nums.size();i++){
if(nums[i]==target)
return i;
else if(nums[i]>target){
return i;
}
}
return nums.size();
}
//改良一
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int len = nums.size();
if(len==0)
return 0;
if(nums[len-1]<target)
return len;
for(int i=0;i<len;i++){
if(nums[i]==target)
return i;
else if(nums[i]>target){
return i;
}
}
return len;
}
};
//二分查找
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int len = nums.size();
if(len==0)
return 0;
if(nums[len-1]<target)
return len;
int left = 0, right = len - 1;
while(left <= right) {
int mid = (left + right) / 2;
if(nums[mid] == target) {
return mid;
}
else if(nums[mid] < target) {
left = mid + 1;
}
else {
right = mid - 1;
}
}
return left;
}
};
暴力法结果
改良一结果
二分查找