社区首页 >专栏 >33. Search in Rotated Sorted Array

33. Search in Rotated Sorted Array

翎野君

发布于 2023-05-12 12:26:31

2040

文章被收录于专栏：翎野君翎野君

两道题

33. Search in Rotated Sorted Array

https://leetcode.com/problems/search-in-rotated-sorted-array/

81. Search in Rotated Sorted Array II

https://leetcode.com/problems/search-in-rotated-sorted-array-ii/

这两道题都是属于循环有序数组的查找问题，无论是查找具体元素的下标还是检查一个目标值是否在数组中，主要的判断逻辑都是一致的。

思路

这种将有序数组翻转成循环有序数组的解决办法是将原数组分段。用首元素start、中间元素mid、尾元素end，可以将数组分为两个子数组s1，s2。

则这两个子数组中，必然有至少一个子数组是有序的。

那么如何确定那一段是有序的呢，通过分析可以看到只有3种情况：

当start就是A中最小的元素时，以下不等式成立：start <= mid <= end
当最小值位于(start, mid]时，则有：mid <= end <= start
当最小值出现在(mid，end]时，则有：end <= start <= mid

所以通过start, mid, end的大小关系，就可以判断出s1和s2哪个是有序的。

通过比较要查找的目标target与start，mid，end的大小关系，就可以知道target位于哪个子数组。

若target位于有序的子数组，则用二分查找就可以了。否则，对无序的子数组重复刚才的过程就可以了。

代码

package com.lingyejun.dating.chap11.toutiao;

public class RotatedArrayQuery {


    /**
     * 33. Search in Rotated Sorted Array
     * 查找目标值下标
     *
     * @param nums
     * @param target
     * @return
     */
    public int searchTargetIndex(int[] nums, int target) {
        int start = 0;
        int end = nums.length - 1;

        while (start <= end) {

            // (start ... middle ... end)
            int mid = (start + end) / 2;

            if (nums[mid] == target) {
                return mid;
            }

            // 前半段数组是有序的
            if (nums[start] < nums[mid]) {
                // target在前半段数组中
                if (target < nums[mid] && target >= nums[start]) {
                    // 将原数组折半，取出前半段数组
                    end = mid - 1;
                } else {
                    // target在后半段数组中
                    start = mid + 1;
                }
            }

            // 后半段数组是有序的
            else if (nums[mid] < nums[end]) {
                // target在后半段数组中
                if (target > nums[mid] && target <= nums[end]) {
                    // 将原数组折半，取出后半段数组
                    start = mid + 1;
                } else {
                    // target在前半段数组中
                    end = mid - 1;
                }
            } else {
                // 此种场景{1, 0, 1, 1, 1} start = mid = end
                start++;
            }
        }
        return -1;
    }

    /**
     * 81. Search in Rotated Sorted Array II
     *
     * 查找是否存在目标值
     *
     * @param nums
     * @param target
     * @return
     */
    public static boolean searchExistsKey(int nums[], int target) {
        int start = 0;
        int end = nums.length - 1;

        while (start <= end) {

            int mid = (start + end) / 2;

            if (nums[mid] == target) {
                return true;
            }

            if (nums[start] < nums[mid]) {

                if (nums[start] <= target && target < nums[mid]) {
                    end = mid - 1;
                } else {
                    start = mid + 1;
                }
            } else if (nums[start] > nums[mid]) {
                if (nums[mid] < target && target <= nums[end]) {
                    start = mid + 1;
                } else {
                    end = mid - 1;
                }

            } else {
                start++;
            }
        }
        return false;
    }

    public static void main(String[] args) {

        int[] array = new int[]{1, 0, 1, 1, 1};

        RotatedArrayQuery caq = new RotatedArrayQuery();

        System.out.println(caq.searchTargetIndex(array, 0));
        System.out.println(caq.searchExistsKey(array, 1));
    }

}