leetcode81 搜索旋转排序数组 II

最新推荐文章于 2025-02-22 19:34:13 发布

原创最新推荐文章于 2025-02-22 19:34:13 发布 · 117 阅读

CC 4.0 BY-SA版权

该博客介绍了如何解决一个特殊的数组查找问题，数组在未知位置进行了旋转。通过实现一个二分查找算法，可以在旋转数组中有效地找到目标值。文章详细讲解了算法思路，包括在旋转点处的特殊情况处理，以及如何根据中间值调整搜索范围。示例展示了算法在具体案例中的应用。

已知存在一个按非降序排列的整数数组 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,4,4,5,6,6,7] 在下标 5 处经旋转后可能变为 [4,5,6,6,7,0,1,2,4,4] 。
给你旋转后的数组 nums 和一个整数 target ，请你编写一个函数来判断给定的目标值是否存在于数组中。如果 nums 中存在这个目标值 target ，则返回 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

提示：
1 <= nums.length <= 5000
-104 <= nums[i] <= 104
题目数据保证 nums 在预先未知的某个下标上进行了旋转
-104 <= target <= 104

思路：
本题采用二分法，可以把nums看做两段递增数列拼接在一起，且后段数列小于前段数列。先通过中值和两端进行比较得出处于那部分数列，再将目标数与中值和两端比较。另外中值与两端数相等的情况需要进行额外考虑。

class Solution {
public:
    bool search(vector<int>& nums, int target) {
        int a,b,m,end;
        a=0;
        b=nums.size()-1;
        while(a<=b)
        {
            m=(a+b)/2;
            if(nums[m]==target) return true;
            if(nums[m]>nums[a])
            {
                if(target<nums[m]&&target>=nums[a])
                {
                    b=m-1;
                }
                else
                {
                    a=m+1;
                }
            }
            else if(nums[m]<nums[b])
            {
                if(target>nums[m]&&target<=nums[b])
                {
                    a=m+1;
                }
                else
                {
                    b=m-1;
                }
            }
            else if(nums[a]==nums[m])
            {
                    a++;
            }
            else
            {
                b--;
            }
        }
        return false;
    }
};