问题
给定旋转数组,寻找目标的下标。
例如数组[4,5,0,1,2,3]是由数组[0,1,2,3,4,5]旋转产生。元素2的索引是4。
找不到指定元素则返回-1.
分析
这是二分查找的一个变体,只是需要加入对边界值的判断。
#include<iostream>
#include<vector>
using namespace std;
int search(vector<int> &nums, int target)
{
int left = 0;
int right = nums.size() - 1;
while (left <= right)
{
int mid = (left + right) / 2;
if (nums[mid] == target)
return mid;
else if (nums[mid] < nums[right]) // 右边是排好序的
{
if (nums[mid] < target && target <= nums[right]) // target在右侧
left = mid + 1;
else
right = mid - 1;
}
else // 左边是排好序的
{
if (nums[mid] > target && target >= nums[left]) // target在左侧
right = mid - 1;
else
left = mid + 1;
}
}
return -1;
}
int main()
{
vector<int> nums{ 4,5,6,0,1,2,3 };
int target = 0;
cin >> target;
int ret = search(nums, target);
cout << "The idx is " << ret << endl;
return 0;
}