leetcode—数组—搜索旋转排序数组

搜索旋转排序数组

题目类型： 数组
题目来源： leetcode传送门

整数数组 nums 按升序排列，数组中的值 互不相同 。

在传递给函数之前，nums 在预先未知的某个下标 k（0 <= k < nums.length）上进行了 旋转，使数组变为 [nums[k], nums[k+1], …, nums[n-1], nums[0], nums[1], …, nums[k-1]]（下标 从 0 开始 计数）。例如， [0,1,2,4,5,6,7] 在下标 3 处经旋转后可能变为 [4,5,6,7,0,1,2] 。

给你 旋转后 的数组 nums 和一个整数 target ，如果 nums 中存在这个目标值 target ，则返回它的下标，否则返回 -1 。

示例1：

输入：nums = [4,5,6,7,0,1,2], target = 0
输出：4

示例2：

输入：nums = [4,5,6,7,0,1,2], target = 3
输出：-1

示例3：

输入：nums = [1], target = 0
输出：-1

提示：

• 1 <= nums.length <= 5000
• -10⁴ <= nums[i] <= 10⁴
• nums 中的每个值都 独一无二
• 题目数据保证 nums 在预先未知的某个下标上进行了旋转
• -10⁴ <= target <= 10⁴

进阶: 你可以设计一个时间复杂度为 O(log n) 的解决方案吗？

这题当然是直接做进阶解法了，不然直接O(N)遍历就太简单了，写起来毫无意义。

看到log n和有序数组，直接想到的就是二分了，但是由于这里有一个旋转的过程，就不能简单的二分。

考虑二分的条件是数组递增或递减，那么这里的数组前半段和后半段都是严格递增的，所以如果我们知道了分界的位置就可以对前半段或后半段二分得到答案了，因为二分是O(log n)的，所以只需要一个O(log n)的方法将分界的位置找出来，那么算法整体就是O(log n)的。

找分界的方法考虑如下：左半部分一定比右半部分的数字都大，这是显然的，那么要想找到分界部分（分界左侧的数字比右侧的数字大）就可以采用双指针的方法，实际上和二分查询的方法类似，这里初始left=0，right=nums.size()-1，每次取mid=(left+right)/2。如果mid小于right，说明mid还是处于右半部分，这时让right=mid；如果mid大于right，说明此时mid在左半部分，需要让left=mid+1，这里如果不加一是无法按照left<right的条件跳出循环的。跳出循环后的left的值即为分界（右半部分第一个位置）。之后只需要判断target在左半部分的区间内还是在右半部分区间内，就可以对相应的部分二分查找了。

思路很简单，但是边界问题还是需要仔细考虑的，改了几次后才把边界问题改好。

代码如下：

int erfen(vector<int>& nums, int target, int left, int right)
    {
        while(left < right)
        {
            int mid = (left + right) / 2;
            if(target <= nums[mid])
            {
                right = mid;
            }
            else
            {
                left = mid + 1;
            }
        }
        if(nums[left] == target)
            return left;
        return -1;
    }

    int search(vector<int>& nums, int target) {
            int left = 0;
            int right = nums.size() - 1;
            int mid = 0;
            while(left < right)
            {
                mid = (left + right) / 2;
                if(nums[mid] > nums[right])
                {
                    left = mid + 1;
                }
                else
                {
                    right = mid;
                }
            }
            mid = left;
            if(target >= nums[mid] && target <= nums[nums.size() - 1])
            {
                return erfen(nums, target, mid, nums.size() - 1);
            }
            else
            {
                return erfen(nums, target, 0, mid - 1);
            }
    }