一、题目
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
二、解析
题目很简单,要在数组中插入目标值,无非是这四种情况。
-
目标值在数组所有元素之前
-
目标值等于数组中某一个元素
-
目标值插入数组中的位置
-
目标值在数组所有元素之后
这四种情况确认清楚了,就可以尝试解题了。
三、暴力法
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int length = nums.size();
int i = 0;
for (;i < length; i++) {
if (nums[i] >= target) {
return i;
}
}
return length;
}
};
四、二分法
大家注意这道题目的前提是数组是有序数组,这也是使用二分查找的基础条件。以后大家「只要看到面试题里给出的数组是有序数组,都可以想一想是否可以使用二分法。同时题目还强调数组中无重复元素,因为一旦有重复元素,使用二分查找法返回的元素下表可能不是唯一的。
class Solution3 {
public:
int searchInsert(vector<int>& nums, int target) {
int length = nums.size();
int left = 0, right = length - 1, ans = length;
while (left <= right) {
int mid = ((right - left) >> 1) + left;
if (target <= nums[mid]) {
ans = mid;
right = mid - 1;
}
else {
left = mid + 1;
}
}
return ans;
}
};
参考:
https://mp.weixin.qq.com/s/fCf5QbPDtE6SSlZ1yh_q8Q