二分查找
leetcode704.二分查找
题目描述
给定一个 n个元素有序的（升序）整型数组 nums 和一个目标值 target ，写一个函数搜索 nums 中的 target，如果目标值存在返回下标，否则返回 -1。
示例 1:
1
2
3
|
输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4
|
示例 2:
1
2
3
|
输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1
|
思路
题目要求
- 在一个升序数组中找到目标元素的下标并返回，
- 若在数组中未找到目标元素，则返回-1。
有序数组中查找元素，考虑使用二分法，也就是折半查找，时间复杂度为O(log n)
注意
- 二分法的使用前提：有序数组
- 二分法的边界条件：
left <= right、left = mid + 1和right = mid - 1
代码
GO
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
func search(nums []int, target int) int {
left := 0
right := len(nums) - 1
for left <= right {
mid := left + (right -left)/2 // 防止直接相加造成溢出
if nums[mid] == target {
return mid
} else if nums[mid] < target {
left = mid + 1 // target 在右区间[mid + 1, right]
} else {
right = mid - 1 // target 在左区间[left, mid - 1]
}
}
return -1
}
|
Link
GitHub