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.

Solution:

public class Solution {
    public int search(int[] nums, int target) {
        int lo=0, hi=nums.length-1;
        while(lo<hi){
            int mid = (lo+hi)/2;
            if(nums[mid]>nums[hi]) lo = mid + 1;
            else hi = mid;
        }
        int shift = lo;
        lo = 0; hi = nums.length-1;
        while(lo<=hi){
            int mid = (lo+hi)/2;
            int real = (mid + shift)%nums.length;
            if(nums[real]==target) return real;
            else if(nums[real]<target) lo = mid +1;
            else hi = mid - 1;
        }
        return -1;
    }
}
comments powered by Disqus