C++旋转数组查找元素

问题

给定旋转数组，寻找目标的下标。
例如数组[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;
}