整数数组 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)
解题思路:
由于题目要求时间复杂度为 O(log n),
因此很容易想到使用二分查找。假设nums数列中最大值的下标为height,则当height不为0或是nums.length-1时,nums数组以height为分界线,下标为0到height的部分为递增数列,假设它为arr1;下标为height+1到nums.length-1的部分也是递增数列,假设其为arr2。因此只要先知道target位于height的左边或是右边,就可以对对应的子数列arr1或arr2使用二分查找查找target是否在arr1或arr2里面。
接下来要判断height的位置同时确定target是在height的左边还是右边。在height不为0或是nums.length-1的情况下,能发现,arr1数组中的述职都不小于nums[0]且都大于nums[nums.length-1];arr2的数值则都小于nums[0]且都不小于nums[height+1],通过以上条件,可以先使用一次二分查找法查找height的值,之后,通过判断target属于arr1或是arr2来在相应的子数组里用二分查找查找target的位置。
代码如下
public int search(int[] nums, int target) {
if (nums.length == 1) {
return nums[0] == target ? 0 : -1;//应对数组长度只有1的情况
}
int height = getHeight(nums);//寻找最高点的下标
if (target >= nums[0]) {
return gerRes(target, nums, 0, height);//往右边进行查找
} else {
return gerRes(target, nums, height + 1, nums.length - 1);
}
}
//查找target的位置
private int gerRes(int target, int[] nums, int left, int right) {
int mid = -1;
while (left <= right) {
mid = left + (right - left) / 2;
if (nums[mid] == target) {
return mid;
} else if (nums[mid] > target) {//往左边找
right = mid - 1;
} else {//往右边找
left = mid + 1;
}
}
return -1;
}
//查找最大值的下标height
private int getHeight(int[] nums) {
int height = -1;
if (nums[0] > nums[nums.length - 1] && nums[0] > nums[1]) {//height==1的情况
height = 0;
} else if (nums[0] < nums[nums.length - 1] && nums[nums.length - 1] > nums[nums.length - 2]) {//height==nums.length-1的情况
height = nums.length - 1;
} else {
int left = 0;
int right = nums.length - 1;
int mid = -1;
while (left <= right) {
mid = left + (right - left) / 2;
if (mid != 0 && mid != nums.length - 1 && nums[mid] > nums[mid - 1] && nums[mid] > nums[mid + 1]) {
height = mid;
break;
} else if (nums[mid] < nums[0]) {//往左边找
right = mid - 1;
} else if (nums[mid] > nums[nums.length - 1]) {//往右边找
left = mid + 1;
}
}
}
return height;
}