假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。
搜索一个给定的目标值,如果数组中存在这个目标值,则返回它的索引,否则返回 -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
解析:本题要求时间复杂度O(logn),因此自然而然想到使用二分法。
本题与普通二分法的不同是原数组经过了一次旋转,因此不满足数组的单调性。
因此而带来的问题是,如何判断在当前的情况下,二分法的左右边界应该如何移动。
设旋转点为R。
根据题目可以得知,经过旋转后,二分法得到的mid可能落在两个部分:0-R段或者R-N段
根据不同的落点,又能继续拓展:
1、nums[m]>nums[0],此时mid落在0-R段
若nums[m]<target,那么target必然落在mid-R段,因此将左边界向右移动,l=m
若nums[m]>target,此时又存在两种可能,落在0-mid段或者R-N段,于是判断target与nums[0]。若target>nums[0],target落在0-mid段,右边界向左移动。反之落在R-N段,左边界向右移动
2、nums[m]<nums[0] ,mid落在R-N段,与上一条相反。
对上述部分进行整理得到
target>nums[mid]>nums[0]
nums[0]>target>nums[mid]
nums[mid]>nums[0]>target
此时移动左边界l=mid
nums[0]>nums[mid]>target
target>nums[0]>nums[mid]
nums[mid]>target>nums[0]
此时移动右边界r=mid
再次尝试简化
对于移动右边界的三条判断进行归纳。
对于以下三条均必满足且只满足两条
①nums[0]>target
②nums[0]>nums[mid]
③target>nums[mid]
此时将这三条判断进行异或结果为0
对于移动左边界的情况,异或结果为1
因此得到了关于二分法边界移动的判断条件
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0;
int right = nums.size()-1;
while(left<right){
int mid = (left+right)>>1;
if(target==nums[mid]) return mid;
if((nums[0]>target)^(nums[0]>nums[mid])^(target>nums[mid])) left = mid+1;
else right = mid-1;
}
return left==right && nums[left]==target?left:-1;
}
};