码迷,mamicode.com
首页 > 其他好文 > 详细

【leetcode】Search in Rotated Sorted Array (hard)

时间:2015-03-20 18:12:38      阅读:149      评论:0      收藏:0      [点我收藏+]

标签:

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.

 

思路:二分搜索,每次去掉一半的错误选项。

注意,每次 l = m + 1, r = m - 1 防止无限循环。

int search(int A[], int n, int target) {
    int l = 0, r = n - 1;
    while(l <= r)  //注意有等号
    {
        int m = (l + r) / 2;
        if(A[m] == target)
            return m;
        if(A[l] <= A[m] && A[m] <= A[r]) //顺序的
        {
            if(A[m] > target)
                r = m - 1;
            else
                l = m + 1;
        }
        else if(A[l] >= A[m] && A[m] <= A[r]) //开头转到了左半部分
        {
            if(A[m] < target && target <= A[r]) //在右半部分
                l = m + 1;
            else
                r = m - 1;
        }
        else //开头转到了右半部分
        {
            if(A[l] <= target && target <= A[m]) //在左半部分
                r = m - 1;
            else
                l = m + 1;
        }
    }
    return -1;
}

 

大神简约版写法:去掉一半选项时的思路不同

int search(int A[], int n,int target) {
    int lo = 0;
    int hi = n - 1;
    while (lo <= hi) {
        int mid = (lo + hi) / 2;
        if (A[mid] == target) return mid;

        if (A[lo] <= A[mid]) {
            if (target >= A[lo] && target < A[mid]) {
                hi = mid - 1;
            } else {
                lo = mid + 1;
            }
        } else {
            if (target > A[mid] && target <= A[hi]) {
                lo = mid + 1;
            } else {
                hi = mid - 1;
            }
        }
    }
    return -1;
}

 

【leetcode】Search in Rotated Sorted Array (hard)

标签:

原文地址:http://www.cnblogs.com/dplearning/p/4353985.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!