题目难度: Medium
原题描述:
Suppose an array sorted in ascending order is rotated at some pivot unknown to you beforehand.
(i.e., 0 1 2 4 5 6 7
might become 4 5 6 7 0 1 2
).
You are given a target value to search. If found in the array return its index, otherwise return -1.
You may assume no duplicate exists in the array.
题目大意:
给你一个递增有序但是在某个位置经过旋转(即某个位置为最大值和最小值的交界处)的数组,并且给你一个目标值,让你在这个数组中查找这个给定值,如果有则返回它在数组的下标,否则返回-1。
解题思路:
一开始看到题目,我就在想直接遍历一次数组O(n)的算法会不会超时,结果写程序试了一下,AC了。但是我知道这道题的原意肯定不是让你去遍历一次数组,不然这道题就没有意思了,更谈不上是Medium的题目。于是我就在想,用二分查找O(log(n))的算法能不能做出这道题。一开始我的想法是,如果能知道数组中最大值和最小值的交界点的位置就好了,这样我就能将数组分成两部分,分别用二分查找去找目标值,但是问题的关键是怎样才能找出这个分界点,而且是要用O(log(n))的时间找出来。
找出分界点的算法仍然可以使用二分查找来解决,我们可以使用下述算法找到最小值元素的位置。对于一个按题目要求给定的数组,我们每次比较数组的最右边的元素和中间的元素,如果最右边的元素大于中间的元素,则它们之间的元素只能是递增的;如果小于,则它们之间的元素是先递增到最大值,然后再从最小值递增到最右边的元素。对于第一种情况,最小值一定在中间及之前的元素里,因此下一次循环时right = mid;对于第二种情况,因为最大和最小值的跳跃处在中间和最右边的元素之间,且中间元素不可能是最小元素,所以下一次循环时left = mid+1。此时由left和right界定的数组仍然属于题目中给定数组的定义,因此就一直这样循环下去,直到left==right时,此时的下标就是最小值元素的下标。
时间复杂度分析:
找出分界点的算法的时间复杂度为O(log(n)),之后两次二分查找的时间复杂度也为O(log(n)),因此总的时间复杂度为O(log(n))。
以下是代码:
int binSearch(int* a , int left , int right , int x)
{
while(left<=right){
int mid = (left+right)/2;
if(x==a[mid]){
return mid;
}
else if(x<a[mid]){
right = mid-1;
}
else if(x>a[mid]){
left = mid+1;
}
}
return -1;
}
int search(int* nums, int numsSize, int target)
{
int left = 0;
int right = numsSize-1;
int minIndex;
while(left<right){
int mid = (left+right)/2;
if(nums[right] > nums[mid]){
right = mid;
}
else{
left = mid+1;
}
}
minIndex = left;
int a = binSearch(nums,0,minIndex-1,target);
int b = binSearch(nums,minIndex,numsSize-1,target);
return (a>b ? a : b);
}