数组-搜索旋转排序数组-中等

描述
假设有一个排序的按未知的旋转轴旋转的数组(比如,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;
	}
};


猜你喜欢

转载自blog.csdn.net/qq_18124075/article/details/80535211