Leetcode中的Search in Rotated Sorted Array II
如果被Rotate的Sorted Array中包含重复的元素,这个问题就变成:
https://leetcode.com/problems/search-in-rotated-sorted-array-ii/
原题
Follow up for “Search in Rotated Sorted Array”:
What if duplicates are allowed?
Would this affect the run-time complexity? How and why?
Write a function to determine if a given target is in the array.
翻译:如果数组中重复的元素,该怎么办?解决方案的时间复杂度是多少?为什么?
问题分析
看看问题I中我们的解决方案
public int search(int[] nums, int target)
{
int low = 0, high = nums.length - 1;
while(low <= high)
{
int mid = (low + high) / 2;
if(nums[mid] == target)
{
return mid;
}
if(nums[mid] >= nums[low])
{
if(target >= nums[low] && target < nums[mid])
{
high = mid - 1;
}
else
{
low = mid + 1;
}
}
else // 注意这个else !
{
if(target <= nums[high] && target > nums[mid])
{
low = mid + 1;
}
else
{
high = mid - 1;
}
}
}
return -1;
}
不难看出,如果mid大于头部元素,那么我们原来的假设满足,这边的else,我们原本可以推出mid小于头部元素(因为没有重复元素),但是如果有重复元素,else只能意味着mid小于等于头部元素。
我们不妨分情况讨论一下,如果mid小于头部元素,原来解法的条件成立;如果等于呢?比如这个数组nums:[1,1,1,1,2,1,1,1,1,1],它是从[1,1,1,1,1,1,1,1,1,2]rotate操作5次得到的。
我们发现nums[mid]==nums[0],但是我们既无法保证low-mid是sorted(看nums),也无法保证mid-high是sorted(如果rotate操作进行7次试试看?)对于上面nums这种情况,坦率的将,不把整个数组遍历一遍,你根本无法知道比如”1.5”到底在不在这个数组中,时间复杂度退化到O(n)。
那么如果mid==low,怎么办呢,我们只能把low位置的元素剔除,继续找。看代码吧,对比上面和下面的代码,一目了然了。
public boolean search(int[] nums, int target)
{
int low = 0, high = nums.length - 1;
while(low <= high)
{
int mid = (low + high) / 2;
if(nums[mid] == target)
{
return true;
}
if(nums[mid] > nums[low])
{
if(target >= nums[low] && target < nums[mid])
{
high = mid - 1;
}
else
{
low = mid + 1;
}
}
else if(nums[mid] < nums[low])
{
if(target <= nums[high] && target > nums[mid])
{
low = mid + 1;
}
else
{
high = mid - 1;
}
}
else // nums[low] == nums[mid] 的情况
{
low = low +1;
}
}
return false;
}
接下来,我们看看在raotate数组中找最小元素的问题。