LeetCode 33搜索旋转数组是二分查找法的一道进阶题目。要求在旋转数组中查找数组索引并把时间复杂度是O(logN)级别。
首先回顾一下二分查找法。二分查找法思路的核心是针对已排序数组,每次找数组中心位置的数,如果比 target 大,就把右指针置到中心位置左边;如果比 target小,就把左指针置到中心位置右边。循环查找直到找到。二分查找法对应的是leetcode 704题。非常基础。
搜索排序旋转数组是把这个已排序数组在某个点旋转了一下。左右部分其实还是有序的。只要多做一个判断就可以。
在旋转之后,前半段数组是顺序的,后半段数组也是顺序的。而数组中间的那个数是否比右边的数小则是不确定的。
所以首先要进行判断,中间数是否比右边的数小,如果小,则说明中间数已经在右半边的数组中了。在此基础上,再判断中间数是否比target小且比右指针大,如果符合,左指针放到中间数右边,否则左指针放到中间数右边。
中间数比右边的数大同理。
思路是正确的,代码写出来可以是这样的:
//这是错误的代码
class Solution {
public int search(int[] nums, int target) {
int left=0,right=nums.length-1;
while(left<=right){
int mid=(left+right)/2;
if(nums[mid]<nums[right]){
if(nums[mid]<target&&target<=nums[right]){
left=mid+1;
}else{
right=mid-1;
}
}else if(nums[mid]>=nums[right]){
if(nums[left]<=target&&target<nums[mid]){
right=mid-1;
}else{
left=mid+1;
}
}else if(nums[mid]==target){
return mid;
}
}
return -1;
}
}
这么写,输出一看,错了。为什么呢?
这里就要引出第二个话题,if 和else if 的先后执行顺序。if 和else if并不是并列执行关系,而是有先后执行顺序。即if执行以后,就不会再执行else if了。如果两个判断条件除了不能同时存在外还有优先级,就要把优先级高的放到if条件里。在实际编码中,很容易忘记这个点,从而导致错误。
对于题目里的官方用例[4,5,6,7,0,1,2],如果按照思路来,在第三次进入while的时候,已经可以求出mid了,但是由于上面的代码返回mid在最后一个if里,程序直接进入了第一个if里进行了错误的判断,再次修改left的值,最终输出的错误的结果。
本质原因是判断条件的逻辑关系不是并列的。返回mid的条件是mid==tartget,而修改指针的第一个条件是nums[mid]>nums[right]。
正确的代码是:
class Solution {
public int search(int[] nums, int target) {
int left=0,right=nums.length-1;
while(left<=right){
int mid=(left+right)/2;
if(nums[mid]==target){//优先返回有结果的时候
return mid;
}else if(nums[mid]<nums[right]){
if(nums[mid]<target&&target<=nums[right]){
left=mid+1;
}else{
right=mid-1;
}
}else{
if(nums[left]<=target&&target<nums[mid]){
right=mid-1;
}else{
left=mid+1;
}
}
}
return -1;
}
}
总结起来就是,如果分析不来的话,就把返回结果优先放到第一个if里。