二分法
题目描述
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
示例1:
输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4
提示:
- 你可以假设 nums 中的所有元素是不重复的。
- n 将在 [1, 10000]之间。
- nums 的每个元素都将在 [-9999, 9999]之间。
题解
思路:
使用二分法查找元素,需要数组满足顺序存储、有序、元素不重复这3个条件。当满足上述条件时可以考虑使用二分法。
二分法有2种写法,一种为左闭右闭,另一种为左闭右开。笔者习惯前者,故采用第一种写法,即区间为 [left, right]。
核心代码【C++】:
int search(vector<int>& nums, int target) {
// nums有序且不重复
int n = nums.size();
int left = 0, right = n - 1;
int middle;
// 二分法,middle指向中间元素
// 习惯用左闭右闭区间,用<=,因为当left == right时,[left,right]仍有效
while(left <= right){
middle = left + ((right - left)/2);//防止溢出
// 本来写的是:middle = (right + left) / 2;
/*
① 该元素=target,直接返回middle
② 该元素<target,[middle + 1,right]
③ 该元素>target,[left,middle - 1]
*/
if(nums[middle] == target) return middle;
else if(nums[middle] < target) left = middle + 1;
else right = middle - 1;
}
return -1;
}
时间复杂度:O(log n)
空间复杂度:O(1)
【参考文档】
- https://programmercarl.com/0704.%E4%BA%8C%E5%88%86%E6%9F%A5%E6%89%BE.html#%E7%AE%97%E6%B3%95%E5%85%AC%E5%BC%80%E8%AF%BE