描述
假设有一个排序的按未知的旋转轴旋转的数组(比如,0 1 2 4 5 6 7 可能成为4 5 6 7 0 1 2)。给定一个目标值进行搜索,如果在数组中找到目标值返回数组中的索引位置,否则返回-1。
你可以假设数组中不存在重复的元素。
样例
给出[4, 5, 1, 2, 3]和target=1,返回 2
给出[4, 5, 1, 2, 3]和target=0,返回 -1
挑战
假设有一个排序的按未知的旋转轴旋转的数组(比如,0 1 2 4 5 6 7 可能成为4 5 6 7 0 1 2)。给定一个目标值进行搜索,如果在数组中找到目标值返回数组中的索引位置,否则返回-1。
你可以假设数组中不存在重复的元素。
样例
给出[4, 5, 1, 2, 3]和target=1,返回 2
给出[4, 5, 1, 2, 3]和target=0,返回 -1
挑战
O(logN) time
分析
这个题目,需要返回目标数字的位置,所以不能进行翻转还原(这样做,会特别麻烦)。我们需要找到那个旋转轴之后的那个数,即在递增过程中,突然递减的那个值。这样可以分成前后两段数组,每一段数组都是递增的。判断目标值在具体哪一段数组里面,同时需要考虑不在这些数组里面的特殊情况。最后,按照二分查找去寻找该段数组里面是否存在目标值。
程序
class Solution { public: /** * @param A: an integer rotated sorted array * @param target: an integer to be searched * @return: an integer */ int find_target(vector<int> &A, int left, int right, int target){ while (right >= left) { int mid = (right + left) >> 1; cout << A[mid] << endl; if (left <= A.size() && A[mid] < target) left = mid + 1; else if (right >= 0 && A[mid] > target) right = mid - 1; else return mid; } return -1; } int search(vector<int> &A, int target) { // write your code here if (A.empty()) return -1; int index, i; for (i = 1; i < A.size(); i++){//寻找到轴点位置,i为翻转之后的开头数字 if (A[i] < A[i - 1]) break; } if (target >= A[0] && target <= A[i - 1])//前半段也是递增的,没有翻转,就直接二分查找 return find_target(A, 0, i - 1, target); else if (target >= A[i] && target <= A[A.size() - 1])//后半段的递增的 return find_target(A, i, A.size() - 1, target); return -1; } };