SearchinRotatedArray2.java

package problems.medium;

import java.util.Arrays;

/**
 * Created by sherxon on 2/11/17.
 */

/**
 * Suppose an array sorted in ascending order is rotated at some pivot unknown to you beforehand.
 * (i.e., 0 1 2 4 5 6 7 might become  3 4 5 6 7 0 1 2 3).
 */
public class SearchinRotatedArray2 {

    /**
     * Solution is easy, we first find rotated pivot index and binary search both sides of pivot.
     * Time complexity is O(N) in the worst case;
     */
    static int search(int[] a, int target) {

        int pivot = findPivot(a); //find rotated point
        System.out.println(pivot);
        int left = Arrays.binarySearch(a, 0, pivot, target); // search left of array
        if (left > -1) return left;
        int right = Arrays.binarySearch(a, pivot, a.length, target); // search right of array
        if (right > -1) return right;
        return -1;
    }

    static int findPivot(int[] a) {
        int lo = 0;
        int hi = a.length - 1;
        while (lo <= hi) {
            while (lo < hi && a[lo] == a[lo + 1]) lo++; // to skip duplicates
            while (hi > lo && a[hi] == a[hi - 1]) hi--; // to skip duplicates
            int mid = lo + (hi - lo) / 2;
            System.out.println(lo + " " + mid + " " + hi);
            if (mid == lo || mid == hi) {
                if (a[lo] < a[hi]) return lo;
                else return hi;
            }
            if (a[mid] > a[hi]) lo = mid;
            else if (a[mid] < a[hi]) hi = mid;
            else return mid;
        }
        return lo;
    }
}