题目
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-in-rotated-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
首先,有序问题,一定是二分法。
如果我们知道“转折点”在哪里,O(lgn)时间复杂度信手拈来。
问题就转变成了O(lgn)的时间复杂度内确定转折点,转折点有什么标志呢?转折点附近一个数是全数组最大,一个数是全数组最小。那么我只需要在O(lgn)时间内用二分法就可以解决问题了。
代码
/*
用两遍二分,第一遍寻找出最小值的位置(flag);
第二遍来进行二分查找;
找最小值位置的目的是:最小值左边(若不是第0位)为最大值,看target是否在nums[0]和nums[flag-1]间,若是,则在0~flag-1区间二分查找,否则在flag~numsSize-1区间查找;(这个过程都是基于flag不在0的基础上的)
*/
int search(int* nums, int numsSize, int target) {
if (numsSize == 0 || numsSize == 1 && nums[0] != target)
return -1;
/*
flag为最小值的位置
t的作用为标志是否旋转数组(即判断flag是否为0),如果没有旋转(为0)那可以跳过寻找最小数阶段
*/
int left = 0, right = numsSize - 1, mid, flag = 0, t = 1;
if (nums[0] < nums[numsSize - 1])
t = 0;
while (left <= right && t) {
mid = left + (right - left) / 2;
if (nums[mid] == target)
return mid;
else {
if (nums[mid] > nums[mid + 1]) {
flag = mid + 1;
break;
}
if (nums[mid-1] > nums[mid]) {
flag = mid;
break;
}
if (nums[mid] > nums[0])
left = mid + 1;
else
right = mid - 1;
}
}
/*
接下来是第二遍二分,要重新判断left和right
*/
if (flag == 0) { //最小值就是第0位
left = 0;
right = numsSize - 1;
}
else if (flag == numsSize - 1) { //最小值是最后一位
if (nums[flag] == target)
return flag;
else {
left = 0;
right = flag - 1;
}
}
else { //其他情况
if (target >= nums[0] && target <= nums[flag - 1]) {
left = 0;
right = flag - 1;
}
else {
left = flag;
right = numsSize - 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;
}
运行结果
一次就直接完美了,真是难以置信。
我参看了官方的题解思路,确实很值得借鉴。
任何一刀切下去,都会转变成一个递增序列和一个旋转序列。只需要判断出目标数在哪一段就可以了,然后递归迭代找出目标数字。
题解的思路我已经理解了,既然我都双百了,那我就不写了嘿嘿。