搜索旋转数组。给定一个排序后的数组,包含n个整数,但这个数组已被旋转过很多次了,次数不详。请编写代码找出数组中的某个元素,假设数组元素原先是按升序排列的。若有多个相同元素,返回索引值最小的一个。
示例1:
输入: arr = [15, 16, 19, 20, 25, 1, 3, 4, 5, 7, 10, 14], target = 5
输出: 8(元素5在该数组中的索引)
示例2:
输入: arr = [15, 16, 19, 20, 25, 1, 3, 4, 5, 7, 10, 14], target = 11
输出: -1 (没有找到)
提示:
点击此处跳转题目。
类似二分查找,但由于优先返回第一次出现的 target,因此主要思想如下:
上述方式类似前序遍历树结构,返回第一个查找的值。
同时,做出如下剪枝优化:
数组被旋转,则数组内所有元素都在区间 [ a r r [ i ] , + ∞ ) ∪ ( − ∞ , a r r [ j ] ] [arr[i], +\infty)\cup(-\infty,arr[j]] [arr[i],+∞)∪(−∞,arr[j]] 内。若 target 不在该区间内,直接返回 -1。
15 , 16 , 19 , 20 , 25 , 1 , 3 , 4 , 5 , 7 , 10 t a r g e t : 12 \begin{aligned} & \sout{15, 16, 19, 20, 25, 1, 3, 4, 5, 7, 10} & target:12 \\ \end{aligned} 15,16,19,20,25,1,3,4,5,7,10target:12
数组未被旋转(有序),则数组内所有元素都在区间 [ a r r [ i ] , a r r [ j ] ] [arr[i], arr[j]] [arr[i],arr[j]] 内。若 target 不在该区间内,直接返回 -1。
1 , 3 , 4 , 5 , 7 , 10 , 15 , 16 , 19 , 20 , 25 t a r g e t : 30 \begin{aligned} & \sout{1, 3, 4, 5, 7, 10, 15, 16, 19, 20, 25} & target:30 \\ \end{aligned} 1,3,4,5,7,10,15,16,19,20,25target:30
mid 查找到值,即 arr[mid] == target。则只查看左枝,减去右枝。
{ 15 , 16 , 19 , 20 , 1 } , 2 ‾ , 3 , 4 , 5 , 7 , 10 t a r g e t : 2 \begin{aligned} & \{15, 16, 19, 20, 1\}, \underline{\bold{2}}, \sout{3, 4, 5, 7, 10} & target:2 \\ \end{aligned} {15,16,19,20,1},2,3,4,5,7,10target:2
右枝有序且 target 在该区间内,则忽略左枝,只看右枝。
15 , 16 , 19 , 20 , 1 , 2 ‾ , { 3 , 4 , 5 , 7 , 10 } t a r g e t : 5 \begin{aligned} & \sout{15, 16, 19, 20, 1}, \underline{2}, \{3, 4, 5, 7, 10\} & target:5 \\ \end{aligned} 15,16,19,20,1,2,{3,4,5,7,10}target:5
public class Solution {
public int Search(int[] arr, int target) {
return Partition(arr, 0, arr.Length - 1, target);
}
public int Partition(int[] arr, int i, int j, int target) {
if (i > j) return -1; // 递归出口
// 剪枝
if (arr[j] < target && target < arr[i]) return -1; // case 1, 直接返回
if (target < arr[i] && target > arr[j]) return -1; // case 2, 直接返回
int mid = (i + j) / 2, left;
if (arr[mid] == target) { // case 3, 减去右半部分
left = Math.Min(Partition(arr, i, mid - 1, target), mid);
return left == -1 ? mid : left;
}
if (arr[mid] < target && target < arr[j]) // case 4, 减去左半部分
return Partition(arr, mid + 1, j, target);
// 优先返回最前面的结果
left = Partition(arr, i, mid - 1, target);
if (left != -1) return left;
return Partition(arr, mid + 1, j, target);
}
}
还可以先用二分查找转折点 k,此时考虑起点为 k 终点为 k - 1 的循环数组,即,从 [k, j] 续上 [i, k - 1] 的有序数组,对其使用二分查找第一个元素。详细代码这里就不附上了。
↱ k 15 , 16 , 19 , 20 , 1 ‾ , 2 , 3 , 4 , 5 , 7 , 10 t a r g e t : 5 ⇓ − , − , − , − , 1 ‾ , 2 , 3 , 4 , 5 , 7 , 10 , 15 , 16 , 19 , 20 t a r g e t : 5 \begin{aligned} & \hspace{5.7em} \Rsh k \\ & 15, 16, 19, 20, \underline{1}, 2, 3, 4, 5, 7, 10 & target:5 \\ & \hspace{5.7em} \Downarrow\\ & -,\ -,\ -,\ -, \hspace{0.1em} \underline{1}, 2, 3, 4, 5, 7, 10, 15, 16, 19, 20 & target:5 \\ \end{aligned} ↱k15,16,19,20,1,2,3,4,5,7,10⇓−, −, −, −,1,2,3,4,5,7,10,15,16,19,20target:5target:5