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

leetcode笔记:Search in Rotated Sorted Array

时间:2015-07-24 18:25:12      阅读:136      评论:0      收藏:0      [点我收藏+]

标签:leetcode   c++   algorithm   sort   binary   

一.题目描述

技术分享

二.解题技巧

由于这道题出现了旋转的情况,即比第一个元素小的元素可能出现在数值的后半段或者不出现。因此,可以考虑采用变种的二分查找,即在比较中间元素与目标之前,先比较第一个元素与目标的关系,这个时候,会出现三种情况:

1.第一个元素刚好等于目标,返回第一个元素的坐标,函数结束;
2.第一个元素大于目标,那么目标就可能存在被旋转到数组后面的情况,这个时候,还要比较与数组中间元素的关系,这个时候又会有三种情况:

a.中间元素大于第一个元素,这个时候,目标可能存在于数组的后半段中,递归调用函数,寻找目标的坐标;
b.中间元素等于目标,返回中间元素的坐标,函数结束;
c.中间元素小于第一个元素,这个时候,又可以分为两种情况进行:

    (1).中间元素小于目标元素,那么目标元素可能存在于数组的后半段中,递归调用函数,寻找目标的坐标;
    (2).中间元素大于目标元素,那么目标元素可能存在于数组的前半段中,递归调用函数,寻找目标的坐标;

3.第一个元素小于目标,这是也有三种情况需要考虑:

a.中间元素等于目标元素,返回中间元素的坐标,函数结束;
b.中间元素大于第一个元素,这个时候,也有两种情况要考虑:

    (1).中间元素大于目标,那么目标元素可能存在于数组的前半段中,递归调用函数,寻找目标的坐标;

    (2).中间元素小于目标,那么目标元素可能存在于数组的后半段中,递归调用函数,寻找目标的坐标;

c.中间元素小于第一个元素,那么目标元素可能存在于数组的前半段中,递归调用函数,寻找目标的坐标;

当然,还需要考虑数组的元素个数为0,1, 2,的情况,以及对于递归的过程中数组的起始位置坐标以及数组中元素的个数,这些才是这道题的难点所在,我也是调试了很久才调通代码的。

三.示例代码

// 时间复杂度O(log n),空间复杂度O(1)
class Solution {
public:
    int search(int A[], int n, int target) {
        int first = 0, last = n;
        while (first != last) {
            const int mid = first + (last - first) / 2;
            if (A[mid] == target)
                return mid;
            if (A[first] <= A[mid]) {
                if (A[first] <= target && target < A[mid])
                    last = mid;
                else
                    first = mid + 1;
            }
            else {
                if (A[mid] < target && target <= A[last - 1])
                    first = mid + 1;
                else
                    last = mid;
            }
        }
        return -1;
    }
};

四.体会
这答题的解题思路不难,难的地方在于边界条件和递归过程中的数组的第一个元素的指针设置和数组元素个数的设置上面,边界条件才是面试题考查的重点啊。

版权声明:本文为博主原创文章,未经博主允许不得转载。

leetcode笔记:Search in Rotated Sorted Array

标签:leetcode   c++   algorithm   sort   binary   

原文地址:http://blog.csdn.net/liyuefeilong/article/details/47043837

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