目录
第一站 LeetCode 新手村
前言
最近玩OJ赛,发现对算法的理解还需要更加扎实,code能力还可以进一步提升,所以做这样一个算法的系列文章,用于记录学习心得,交流经验,更好地进步和成长。
35. 搜索插入位置
题目描述
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
示例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 <= 104
-104 <= nums[i] <= 104
nums 为 无重复元素 的 升序 排列数组
-104 <= target <= 104
解题思路
预知
LeetCode是核心代码模式,所以只需要考虑核心算法,输入由系统自动完成,最后的输出以return返回;
思路
由于,有了时间复杂的为O(log n)的要求,所以采用二分法;
这个题直接暴力遍历也可以过
具体思路请参考代码和注释
代码
C++ 二分 时间复杂度O(log n)
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
// 二分查找 时间复杂度 O(log n)
int n = nums.size();
int left = 0, right = n-1, ans = n;
while(left <= right){ //用右移可以防止越界,右移一位相当于除以2
int mid = ((right-left) >> 1 ) + left; // (left+right)/2 = left + (right-left)/2
if(target <= nums[mid]){
ans = mid;
right = mid-1;
}else{
left = mid+1;
}
}
return ans;
}
};
暴力
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
for(int i=0;i<nums.size();i++){
if(nums[i]>=target){
return i;
}
}
return nums.size();
}
};
总结
以上就是今天要讲的内容,本文仅仅简单讲解了《35. 搜索插入位置》这一题目,对二分法的使用进一步熟悉。
题目来源
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/search-insert-position