1.题目
假设按升序排序的数组在事先未知的某个点处旋转。(即[0,1,2,4,5,6,7]可能变为[4,5,6,7,0,1,2])。
您将获得要搜索的目标值。 如果在数组中找到则返回其索引,否则返回-1。
假设数组中不存在重复。
要求算法的运行时复杂度必须为O(log n)。
示例:
Input: nums = [4,5,6,7,0,1,2], target = 0 Output: 4
2.思路
旋转数组被mid分割后有两种可能性
(1) nums[lo, mi]是有序的,如下图所示。
此时有两种可能性:
i)target在nums[lo, mi) 可深入其中继续查找(hi = mid - 1)。
ii)否则,若目标元素存在,target在nums(mi, hi]中 可深入其中继续查找(lo = mid + 1)。
(2)nums[mi, hi]是有序的,如下图所示。
此时同样有两种可能性
i)target在nums(mi, hi] 可深入其中继续查找(lo = mid + 1)。
ii)否则,若目标元素存在,target在num[lo, mi)s中 可深入其中继续查找(hi = mid - 1)。
折半缩小查找范围。
3.实现
class Solution {
public:
int search(vector<int>& nums, int target) {
if(nums.empty())
return -1;
int lo = 0;
int hi = nums.size() - 1;
while(lo <= hi)
{
int mi = lo + (hi - lo) / 2;
if(nums[mi] == target)
return mi;
if(nums[mi] > nums[hi]) //第一种情况
{
if(nums[lo] <= target && target < nums[mi])
hi = mi - 1; //左边有序区域
else
lo = mi + 1;
}
else //第二种情况
{
if(nums[mi] < target && target <= nums[hi])
lo = mi + 1;
else
hi = mi - 1;
}
}
return -1;
}
};
注:
解释一下if条件中的“=”。
if(nums[lo] <= target && target < nums[mi])
if(nums[mi] < target && target <= nums[hi])
解释一下这两条逻辑
在二分查找中,经过不断移动左右边界,一般会退化到只剩下两个元素的情况,因此可以用两个元素的情况做测试。
例如[2, 3]之中查找3
lo = 0 hi = 1 mi = 0 是第二种情况。
如果if语句是if(nums[mi] < target && target < nums[hi])
那会进入else中,因此target < nums[hi] 应改为 target <= nums[hi]
同理在例如[3, 2]之中查找3
lo = 0 hi = 1 mi = 0 是第一种情况。
如果if语句是 if(nums[lo] < target && target < nums[mi])
那会进入else中,因此nums[lo] < target 应改为 nums[lo] <= target