题目
整数数组 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) 的算法解决此问题。
示例
输入:nums = [4,5,6,7,0,1,2], target = 0
输出:4
解析
这道题首先题意上是可以理解的,前半段是一个递增数组,后半段也是一个递增数组。然后解决思路是二分查找,注释都在代码里了,直接看代码:
func search(nums []int, target int) int {
// 左闭右开写法
left := 0
right := len(nums)
// 先提前判断target在左子数组中还是右子数组
flag := false // 在左子数组
if nums[0] > target { // 目标在右子数组(前半段的递增数组中,第一个元素就大于target,说明和target相等的下标在右子数组中)
flag = true
}
for left < right {
mid := left + (right-left)/2
if nums[mid] == target { // 直接找到了,就返回完事
return mid
}
if flag { // 目标在右子数组
if nums[mid] > nums[0] { // 目标都在右子数组了,中点还在左子数组呢,赶紧处理
left = mid + 1
} else { // 中点在右子数组
if nums[mid] < target {
left = mid + 1
} else {
right = mid
}
}
} else { // 目标在左子数组(和上方同理)
if nums[mid] < nums[0] { // 中点在右子数组
right = mid
} else { // 中点在左子数组
if nums[mid] < target {
left = mid + 1
} else {
right = mid
}
}
}
}
return -1
}