题目相关
题目链接
LeetCode中国,https://leetcode-cn.com/problems/search-insert-position/。
题目描述
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例
示例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
题目分析
LeetCode 给出本题难度简单。
一个标准的二分查找模板题,没有什么需要特别说明的地方。可以使用 STL 来实现,也可以使用自己写一个二分查找来实现。
AC 参考代码
STL 模板
使用 STL 模板只需要调用 lower_bound() 即可。
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
return lower_bound(nums.begin(), nums.end(), target) - nums.begin();
}
};
自己实现
只需要使用二分查找模板就可以了。
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int left = 0;
int right = nums.size();
int mid;
while (left < right) {
mid = left + ((right-left)>>1);
if (nums[mid]<target) {
left = mid+1;
} else {
right = mid;
}
}
return left;
}
};
惭愧,自己实现的代码效率还是比不过 STL 模板。