2014年3月16日星期日

LeetCode - Search in Rotated Sorted Array



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.


public class Solution {
    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<=end) {
            int mid = (start+end)/2;
            if(A[mid]==target) {
                return mid;
            } else if(A[mid]<target) {
                if(A[start]<A[end] || A[start]<A[mid] || target<=A[end]) {
                    start = mid + 1;
                } else {
                    end = mid -1;
                }
            } else {
                if(A[start]<A[end] || A[mid]<A[end] || target>=A[start]) {
                    end = mid - 1;
                } else {
                    start = mid + 1;
                }
            }
        }
        return -1;
    }
}


public class Solution {
    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<=end) {
            int mid = (start+end)/2;
            if(A[mid]==target) {
                return mid;
            }
            // left is sorted
            if(A[start]<=A[mid]) {
                if(target<A[mid] && target>=A[start]) {
                    end = mid - 1;
                } else {
                    start = mid + 1;
                }
            } else {
                if(target>A[mid] && target<=A[end]) {
                    start = mid + 1;
                } else {
                    end = mid - 1;
                }
            }
        }
        return -1;
    }
}

没有评论:

发表评论