33. Search in Rotated Sorted Array

最新推荐文章于 2020-02-22 15:07:37 发布

原创最新推荐文章于 2020-02-22 15:07:37 发布 · 214 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

58 篇文章

订阅专栏

本文介绍了一种在旋转排序数组中查找目标元素的高效算法。通过两种方法实现：一是利用STL map进行查找；二是采用改进的二分查找，达到O(logN)的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

假设按升序排序的数组在未知的某处旋转。（即[0,1,2,4,5,6,7]可能变为[4,5,6,7,0,1,2]）。使用时间复杂度为O(logN)的方法查询数组内有没有target。

思路1：利用stl的map

int search(vector<int>& nums, int target) {//利用map，99%
	map<int,int> a, b;
	int temp = INT32_MIN;
	int i = 0, sz = nums.size();
	if (!sz)
		return -1;
	auto it = a.begin();
	for (; nums[i] > temp; ++i) {
		a.insert(pair<int, int>(nums[i], i));
		temp = nums[i];
	}
	for (; i < sz; ++i) {
		b.insert(pair<int, int>(nums[i], i));
	}

	if (target >= a.begin()->first ){
		if ((it = a.find(target)) != a.end())
			return it->second;
		else
			return -1;
	}
	else {
		if ((it = b.find(target)) != b.end())
			return it->second;
		else
			return -1;
	}
}

思路2：二分查找

int search(vector<int>& nums, int target) {//二分法
	int l = 0, r = nums.size() - 1, mid;
	while (l <= r) {
		mid = (r + l) / 2;
		if (nums[mid] == target)
			return mid;
			
		else if (nums[mid] < nums[r]) {//旋转位置在mid左边
			if (nums[mid] < target && target <= nums[r])
				l = mid + 1;
			else
				r = mid - 1;
		}
		
		else {//旋转位置在mid右边
			if (nums[mid] > target && target >= nums[l])
				r = mid - 1;
			else
				l = mid + 1;
		}
	}
	return -1;
}