搜索插入位置
- 题解1 二分查找
- 防越界写法
- 题解2 STL大法
- 两行
给定一个排序数组和一个目标值,在数组中找到目标值, 并返回其索引。如果目标值不存在于数组中, 返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O ( l o g n ) O(log n) O(logn)的算法。
示例 1:
输入: nums = [1,3,5,6], target = 5
输出: 2
示例 2:
输入: nums = [1,3,5,6], target = 2
输出: 1
示例 3:
输入: nums = [1,3,5,6], target = 7
输出: 4
提示:
- 1 <=
nums.length
<= 1 0 4 10^4 104 - − 1 0 4 -10^4 −104 <=
nums[i]
<= 1 0 4 10^4 104 nums
为 无重复元素 的 升序 排列数组- − 1 0 4 -10^4 −104 <=
target
<= 1 0 4 10^4 104
题解1 二分查找
class Solution {
public:int searchInsert(vector<int>& nums, int target) {int len = nums.size();int left = 0;int right = len-1;int pos = 0;while(left <= right){int mid = (left+right) >> 1;if(nums[mid] == target) return mid;else if(nums[mid] < target){left = mid+1;// left就是升序情况下 应该插入的位置pos = left;}else{right = mid-1;}}return pos;}
};
防越界写法
class Solution {
public:int searchInsert(vector<int>& nums, int target) {int n = nums.size();int left = 0, right = n - 1, ans = n;while (left <= right) {int mid = ((right - left) >> 1) + left;if (target <= nums[mid]) {// mid = left + (difference)>>1 (Key: 找到第一个下标,对应值 >= target)ans = mid;right = mid - 1;} else {left = mid + 1;}}return ans;}
};
题解2 STL大法
class Solution {
public:int searchInsert(vector<int>& nums, int target) {auto it = find(nums.begin(),nums.end(),target);if(it!=nums.end()){return it-nums.begin();}auto first =lower_bound(nums.begin(), nums.end(), target);return first-nums.begin();}
};
两行
class Solution {
public:int searchInsert(vector<int>& nums, int target) {auto st = nums.cbegin(), ed = nums.cend();return lower_bound(st, ed, target) - st;}
};