Search in Rotated Sorted Array
- leetcode: Search in Rotated Sorted Array | LeetCode OJ
- lintcode:
(i.e., 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.
Example
For [4, 5, 1, 2, 3]
and target=1
, return 2
.
For [4, 5, 1, 2, 3]
and target=0
, return -1
.
Challenge
题解 - 找到有序数组
对于旋转数组的分析可使用画图的方法,如下图所示,升序数组经旋转后可能为如下两种形式。
public class Solution {
/**
*@param A : an integer rotated sorted array
*@param target : an integer to be searched
*return : an integer
*/
public int search(int[] A, int target) {
int lb = 0, ub = A.length - 1;
while (lb + 1 < ub) {
int mid = lb + (ub - lb) / 2;
if (A[mid] == target) return mid;
if (A[mid] > A[lb]) {
// case1: numbers between lb and mid are sorted
if (A[lb] <= target && target <= A[mid]) {
ub = mid;
} else {
lb = mid;
}
} else {
// case2: numbers between mid and ub are sorted
if (A[mid] <= target && target <= A[ub]) {
lb = mid;
} else {
}
}
}
if (A[lb] == target) {
return lb;
} else if (A[ub] == target) {
return ub;
}
return -1;
}
}
- 若
target == A[mid]
,索引找到,直接返回 - 寻找局部有序数组,分析
A[mid]
和两段有序的数组特点,由于旋转后前面有序数组最小值都比后面有序数组最大值大。故若A[start] < A[mid]
成立,则start与mid间的元素必有序(要么是前一段有序数组,要么是后一段有序数组,还有可能是未旋转数组)。 - 接着在有序数组
A[start]~A[mid]
间进行二分搜索,但能在A[start]~A[mid]
间搜索的前提是。 - 接着在有序数组
A[mid]~A[end]
间进行二分搜索,注意前提条件。 - 搜索完毕时索引若不是mid或者未满足while循环条件,则测试A[start]或者A[end]是否满足条件。
- 最后若未找到满足条件的索引,则返回-1.
分两段二分,时间复杂度仍近似为 O(\log n).