一、题目
整数数组 nums 按升序排列,数组中的值 互不相同 。
在传递给函数之前,nums 在预先未知的某个下标 k(0 <= k < nums.length)上进行了 旋转,使数组变为 [nums[k], nums[k+1], ..., nums[n-1], nums[0], nums[1], ..., nums[k-1]](下标 从 0 开始 计数)。例如, [0,1,2,4,5,6,7] 在下标 3 处经旋转后可能变为 [4,5,6,7,0,1,2] 。
给你 旋转后 的数组 nums 和一个整数 target ,如果 nums 中存在这个目标值 target ,则返回它的下标,否则返回 -1 。
示例 1:
输入:nums = [4,5,6,7,0,1,2], target = 0
输出:4
示例 2:
输入:nums = [4,5,6,7,0,1,2], target = 3
输出:-1
示例 3:
输入:nums = [1], target = 0
输出:-1
提示:
1 <= nums.length <= 5000
-10^4 <= nums[i] <= 10^4
nums 中的每个值都 独一无二
题目数据保证 nums 在预先未知的某个下标上进行了旋转
-10^4 <= target <= 10^4
二、思路
- 虽然将有序数组进行旋转了,但是依旧能有一部分是有序的。我们将数组分为[l, mid] 和[mid + 1, r]两个部分,可以看出必然有一部分是有序的,那么我们只需要判断target是否在这有序的部分,若不在继续对另一部分进行下一次分割,直到找到target元素或者循环终止。
三、代码
class Solution {
public:
int search(vector<int>& nums, int target) {
int l = 0, r = nums.size() - 1, mid;
//分成[l, mid] [mid + 1, r] 必定有一个有序,若有序则判断target是否在其中若在继续遍历
while (l < r) {
mid = (l + r) / 2;
if (nums[l] <= nums[mid]) {
//[l, mid]有序
if (nums[l] <= target && target <= nums[mid] ) {
r = mid;
} else {
l = mid + 1;
}
} else {
//[mid+1, r]有序
if (nums[mid + 1] <= target && target <= nums[r]) {
l = mid + 1;
} else {
r = mid;
}
}
//printf("%d %d\n", l, r);
}
return nums[r] == target? r : -1;
}
};