33. 搜索旋转排序数组
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。
搜索一个给定的目标值,如果数组中存在这个目标值,则返回它的索引,否则返回 -1 。
你可以假设数组中不存在重复的元素。
你的算法时间复杂度必须是 O(log n) 级别。
示例 1:
输入: nums = [4,5,6,7,0,1,2], target = 0
输出: 4
示例 2:
输入: nums = [4,5,6,7,0,1,2], target = 3
输出: -1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-in-rotated-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题
部分有序数组查找+复杂度logn,用二分查找;
旋转排序数组对半分,一定有一半是有序的;
判断target是否在有序的那一半中,若是,在那一半中查找;
若否,在另外一半中查找;
注意点
若nums为空数组,返回-1;
若nums只有一个元素,判断这个元素即可;
class Solution {
public:
int search(vector<int>& nums, int target) {
int n=nums.size();
if(!n) return -1;
if(n==1) return nums[0]==target? 0:-1;
int l=0,r=n-1;
int mid;
while(l<=r)
{
mid=l+(r-l)/2;
if(nums[mid]==target) return mid;
if(nums[l]<=nums[mid]) //左边有序
if(target>=nums[l]&&target<nums[mid])
r=mid-1;
else
l=mid+1; //看右边
else //左边无序。右边有序
if(target>nums[mid]&&target<=nums[r])
l=mid+1;
else
r=mid-1;
}
return -1;
}
};
二分查找设计较多边界细节;
while循环条件为l<=r;
若中点为所找值——返回中点下标;
若中点左边为有序数组:判断target是否在左边的范围内,是则搜索左边(r=mid-1),否则搜索右边(l=mid+1);
若中点右边为有序数组:判断target是否在右边,是则搜索右边(l=mid+1),否则搜索左边(r=mid-1);
若中点左右都为有序数组,也会先判断左边是否含中点,否则搜索右边,直到l>r,返回-1;
关键:让mid遍历可能范围内的每一个点
81. 搜索旋转排序数组 II
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,0,1,2,2,5,6] 可能变为 [2,5,6,0,0,1,2] )。
编写一个函数来判断给定的目标值是否存在于数组中。若存在返回 true,否则返回 false。
示例 1:
输入: nums = [2,5,6,0,0,1,2], target = 0
输出: true
示例 2:
输入: nums = [2,5,6,0,0,1,2], target = 3
输出: false
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-in-rotated-sorted-array-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题
与上一题相比,列表中多了重复元素;
如何二分?
1.若有重复元素,则nums[l]==nums[mid],无法分清左右哪个是有序数列,则l++;
2.nums[I]<nums[mid],说明左边是有序数列,判断target是否在左边,若在则r=mid-1,进行二分,若不在则l=mid+1,向右二分;
3.nums[l]>nums[mid],说明右边是有序数列,判断target是否在右边,若在则l=mid+1,进行二分,若不在则r=mid-1,向左二分;
class Solution {
public:
bool search(vector<int>& nums, int target) {
int n = nums.size();
if(!n) return false;
if(n==1) return nums[0]==target? true:false;
int l=0;
int r=n-1;
int mid;
while(l<=r)
{
mid = l+(r-l)/2;
if(nums[mid]==target) return true;
if(nums[l]==nums[mid])
{
l++;
continue;
}
if(nums[l]<nums[mid])
if(target<nums[mid]&&nums[l]<=target)
r=mid-1;
else
l=mid+1;
else
if(target<=nums[r]&&nums[mid]<target)
l=mid+1;
else
r=mid-1;
}
return false;
}
};
这是把=<>三种分开来写;
写法2:
l和r略过相同元素,再进行二分查找;
nums[l]<=nums[mid],若等于,说明l和mid相同,
若此时l==r,则判断后r和l之一改变,推出循环,返回false;
若此时l=r-1,则mid=l,判断后l=mid+1,判断r,若也不为target则返回false;
class Solution {
public:
bool search(vector<int>& nums, int target) {
int n = nums.size();
if(!n) return false;
if(n==1) return nums[0]==target? true:false;
int l=0;
int r=n-1;
int mid;
while(l<=r)
{
while(l<r&&nums[l]==nums[l+1]) ++l;
while(l<r&&nums[r]==nums[r-1]) --r;
mid = l+(r-l)/2;
if(nums[mid]==target) return true;
if(nums[l]<=nums[mid])
if(target<nums[mid]&&nums[l]<=target)
r=mid-1;
else
l=mid+1;
else
if(target<=nums[r]&&nums[mid]<target)
l=mid+1;
else
r=mid-1;
}
return false;
}
};
总结
二分查找要注意边界细节,写之前把剩余1-2个数的情况推倒一遍;