Search in Rotated Sorted Array

Lintcode
題意:

解題思路:

暴力法:掃一次陣列即可,花 O(N)。

二分法:花 O(logN)

類似Find Minimum in Rotated Sorted Array 的作法,但要分兩個情況(1,2)來處理

1:start < mid

2:start >= mid

接著再判斷 target是在哪個區間
若是第一個情況的話,則比較target是否在 start 與 mid之間。

  1. 是的話,則把 end 移到 mid
  2. 否的話,把 start 移到 mid

若是第二個情況的話,則比較target是否在mid與 end之間。

  1. 是的話,則把start 移到 mid
  2. 否的話,把 end 移到 mid

每一輪作完仍是一個縮減版的滾動陣列,便繼續往下作。

由於這模板是把搜尋範圍縮短到兩個值,最後記得跟 target 比較再決定輸出,其程式碼如下:

public class Solution {
    /** 
     *@param A : an integer rotated sorted array
     *@param target :  an integer to be searched
     *return : an integer
     */
    public int search(int[] A, int target) {

        if (A == null || A.length == 0) {
            return -1;
        }

        int start = 0; 
        int end = A.length - 1;
        while (start + 1 < end) {
            int mid = start + (end - start) / 2;
            if (target == A[mid]) {
                return mid;
            }

            if (A[mid] < A[end]) {
                if (target >= A[mid] && target <= A[end]) {
                    start = mid;
                } else {
                    end = mid;
                }
            } else {
                if (target >= A[start] && target <= A[mid]) {
                    end = mid;
                } else {
                    start = mid;
                }
            }
        }

        if (A[start] == target) {
            return start;
        } else if (A[end] == target) {
            return end;
        } else {
            return -1;
        }
    }
}
[Updated on 2019.10.01]

網友leetcode.wang 給了以下很好的說明

参考这里,算法基于一个事实,数组从任意位置劈开后,至少有一半是有序的,什么意思呢?

比如 [ 4 5 6 7 1 2 3] ,从 7 劈开,左边是 [ 4 5 6 7] 右边是 [ 7 1 2 3],左边是有序的。

基于这个事实。

我们可以先找到哪一段是有序的 (只要判断端点即可),然后看 target 在不在这一段里,如果在,那么就把另一半丢弃。如果不在,那么就把这一段丢弃。

class Solution {
    public int search(int[] nums, int target) {
        if (nums == null || nums.length == 0) {
            return -1;
        }

        int start = 0;
        int end = nums.length - 1;

        while (start <= end) {
            int mid = start + (end - start) / 2;
            if (nums[mid] == target) {
                return mid;
            }
            // 左半邊是有序的
            if (nums[mid] >= nums[start]) {
                // target 在這一段
                if (target >= nums[start] && target < nums[mid]) {
                    end = mid -1;
                } else {
                    start = mid + 1;
                }
            } else { // 右半是有序的
                if (target > nums[mid] && target <= nums[end]) {
                    start = mid + 1;
                } else {
                    end = mid - 1;
                }
            }
        }

        return -1;
    }
}

Reference

  1. http://www.jiuzhang.com/solutions/search-in-rotated-sorted-array/
  2. https://leetcode.wang/leetCode-33-Search-in-Rotated-Sorted-Array.html

results matching ""

    No results matching ""