Search in Rotated Sorted Array

最新推荐文章于 2020-08-05 08:57:53 发布

原创最新推荐文章于 2020-08-05 08:57:53 发布 · 3.1k 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

46 篇文章

订阅专栏

本文介绍了一种在旋转数组中查找特定元素的算法。通过两次二分查找，首先找到旋转点，然后确定目标值所在的子数组并进行搜索。适用于不包含重复元素的有序旋转数组。

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

难度：2

Suppose a sorted array is rotated at some pivot unknown to you beforehand.

(i.e., 0 1 2 4 5 6 7 might become 4 5 6 7 0 1 2).

You are given a target value to search. If found in the array return its index, otherwise return -1.

You may assume no duplicate exists in the array.

嗯。。拆分两段后，容易发现第一段的最后一个数大于第二段的最后一个数，并且第二段的第一个数小于第二段的最后一个数

比如

4 5 6 7 0 1 2

2一定小于7并且大于0

任何题意给定的序列都满足这个关系

所以先二分找到两段的交点

然后分别二分找到答案

class Solution
{
public:
    int search_2(int A[], int L, int R, int target)
    {
        while(L<=R)
        {
            int mid=(L+R)>>1;
            if(A[mid]>target)
            {
                R=mid-1;
            }
            else if(A[mid]<target)
            {
                L=mid+1;
            }
            else    return mid;
        }
        return -1;
    }
    int search(int A[], int n, int target)
    {
        if(n == 0)  return -1;
        if(n == 1)
        {
            if(A[0] == target)  return 0;
            else    return -1;
        }
        if(n == 2)
        {
            if(A[0] == target)  return 0;
            else if(A[1] == target) return 1;
            else    return -1;
        }
        int L=0,R=n-2;
        while(L<R)//[0,n-2]找一个比A[n-1]大的数
        {
            int mid=(L+R>>1)+1;
            if(A[mid]>A[n-1])   L=mid;
            else  R=mid-1;
        }
        int split=L;
        if(A[L]<A[n-1]) split=n-1;
        //[0,split],[split+1,n-1]
        int ans;
        ans=search_2(A,0,split,target);
        if(ans!=-1) return ans;
        if(split+1<=n-1)
        {
            return search_2(A,split+1,n-1,target);
        }
        return -1;
    }
};