33. Search in Rotated Sorted Array
...大约 1 分钟
整数数组
nums
按升序排列,数组中的值 互不相同 。在传递给函数之前,
nums
在预先未知的某个下标k
(0 <= k < nums.length
)上进行了 旋转,使数组变为[nums[k], nums[k+1], ..., nums[n-1], nums[0], nums[1], ..., nums[k-1]]
(下标 从 0 开始 计数)。例如,[0,1,2,4,5,6,7]
在下标3
处经旋转后可能变为[4,5,6,7,0,1,2]
。给你 旋转后 的数组
nums
和一个整数target
,如果nums
中存在这个目标值target
,则返回它的下标,否则返回-1
。你必须设计一个时间复杂度为
O(log n)
的算法解决此问题。示例 1:
输入:nums = [4,5,6,7,0,1,2], target = 0 输出:4
示例 2:
输入:nums = [4,5,6,7,0,1,2], target = 3 输出:-1
示例 3:
输入:nums = [1], target = 0 输出:-1
提示:
1 <= nums.length <= 5000
-10^4 <= nums[i] <= 10^4
nums
中的每个值都 独一无二- 题目数据保证
nums
在预先未知的某个下标上进行了旋转-10^4 <= target <= 10^4
1. 二分查找
TC: O(logn), SC: O(1)
找出分界点, 对两个区间分别进行二分查找
func search(nums []int, target int) int {
// 找出分界点
n := len(nums)
idx := -1
for i := 0; i < n-1; i++ {
if nums[i] > nums[i+1] {
idx = i
}
}
// 查找区间
res1 := binarySearch(nums, 0, idx, target)
res2 := binarySearch(nums, idx+1, n-1, target)
if res1 > -1 {
return res1
}
return res2
}
func binarySearch(nums []int, left, right, target int) int {
for left <= right {
mid := (left+right) / 2
midVal := nums[mid]
if target == midVal {
return mid
} else if target > midVal {
left = mid + 1
} else {
right = mid - 1
}
}
return -1
}
Reference
Powered by Waline v2.15.2