Idiot-maker

  :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::

https://oj.leetcode.com/problems/find-minimum-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).

Find the minimum element.

You may assume no duplicate exists in the array.

解题思路:

最朴素的O(n)的方法,就是找到突然变小的那个元素,肯定是最小的。否则就说明所有元素一直变大,本来就是排序的,则返回第一个元素。

public class Solution {
    public int findMin(int[] num) {
        if(num.length == 0){
            return 0;
        }
        
        int min = num[0];
        
        for(int i = 0; i < num.length; i++){
            if(num[i] < min){
                return num[i];
            }
        }
        return min;
    }
}

 不过本题还有一个二分查找的方法,可以将时间复杂度降为O(lgN)。代码如下。

public class Solution {
    public int findMin(int[] num) {
        if(num.length == 0){
            return 0;
        }
        
        if(num.length == 1){
            return num[0];
        }
        
        int start = 0;
        int end = num.length - 1;
        int min = Integer.MAX_VALUE;
        
        if(num[start] < num[end]){
            return num[start];
        }
        
        while(start <= end){
            if(start == end){
                return Math.min(min, num[start]);
            }
            int mid = (start + end) / 2;
            
            if(num[start] <= num[mid]){
                min = Math.min(min, num[start]);
                start = mid + 1;
            }else{
                min = Math.min(min, num[mid]);
                end = mid - 1;
            }
        }
        return min;
    }
}

二分查找是一个很简单,却很难完全写对的算法,编程珠玑里写道,差不多只有10%的程序员能正确的实现它。问题主要处在边界变量的定义,和循环条件的定义上。让我们从程序的开头看起。

将end定义为length - 1,就意味着这是一个[0, length - 1]的左右都闭合的区间,所以循环的条件一定是start <= end。所以再到循环内部,start = mid + 1, end = mid - 1,都是因为这是一个闭合的区间,这样做才不会漏元素。

如果end定义为length,那么就意味着这个区间是左闭合有开的,[0, length)。所以后面while的条件就变为start < end,循环内部,start = mid + 1, end = mid。

这样理解起来就不会出错。其实这就是循环不变式的概念,loop invariant。可以参考下面的页面,对于这个比较基础的知识介绍的很好。

http://www.cppblog.com/converse/archive/2009/09/21/96893.aspx

update: 2015/03/11

下面一段我写的,更为简洁的代码。看几个例子,7 8 0 1 2 3 4 5 6,mid=2;3 4 5 6 7 8 0 1 2,mid=7。

mid > end的时候,代表前面是排序的,可能折断点在后面,于是start=mid+1。

mid < end的时候,代表后面是排序的,可能折断点在前面,但是这时候不能end = mid - 1。为什么?看一个例子,4 5 0 1 2。这时mid < end,但是mid就是折断点,所以这时end = mid。

mid == end的时候,说明当前mid就是最小值。

为什么mid要和end比较,不能和start?因为除非在start==end的情况,否则mid不可能等于end,但是start可能等于mid,比如只有两个元素,[2,1],这时start==mid<end,所以返回的是错误的2。

另外还要注意一点。while内部更新start和end的各个条件要符合互相没有交集,但是合起来又可以组成全集的原理。因为前半句违背了,可能一次更新多次start和end,这样有些范围就跳过不搜索了。后半句违背了,可能使得start和end得不到更新,引起死循环。

这个性质其实也和上面说的循环不变式同理,需要体会。

public class Solution {
    public int findMin(int[] num) {
        int start = 0;
        int end = num.length - 1;
        while(start <= end){
            int mid = (start + end) / 2;
            if(num[end] == num[mid]){
                return num[mid];
            }
            if(num[end] < num[mid]){
                start = mid + 1;
            }
            if(num[end] > num[mid]){
                end = mid;
            }
        }
        return -1;
    }
}

 二分查找的题目很多,基本原理就是这样,但是具体情况,mid的值和哪个值比较,状态变化和终止的条件,变化万千,需要细细体会。

这道题是二分查找里比较经典的题目,也比较灵活,相应条件的定义和start、end的转换可以有多种写法。

posted on 2015-02-01 23:04  NickyYe  阅读(706)  评论(1编辑  收藏  举报