[LintCode]寻找峰值

题目:

你给出一个整数数组(size为n),其具有以下特点:

  • 相邻位置的数字是不同的
  • A[0] < A[1] 并且 A[n - 2] > A[n - 1]

假定P是峰值的位置则满足A[P] > A[P-1]A[P] > A[P+1],返回数组中任意一个峰值的位置。

您在真实的面试中是否遇到过这个题?  
Yes
样例

给出数组[1, 2, 1, 3, 4, 5, 7, 6]返回1, 即数值 2 所在位置, 或者6, 即数值 7 所在位置.

注意

数组可能包含多个峰值,只需找到其中的任何一个即可

解答:直接查找太简单,时间复杂度为O(n)。可用二分查找,时间复杂度为O(lgn)

代码如下:

public int findPeak(int[] A) {
        int begin = 0;
        int end = A.length - 1;
        while(begin < end) {
            int mid = begin + (end - begin)/2;
            if(mid == 0)
                return 1;
            if(mid == A.length - 1)
                return mid - 1;
            if(A[mid] > A[mid - 1] && A[mid] > A[mid + 1])
                return mid;
            else if(A[mid] < A[mid - 1])
                end = mid - 1;
            else
                begin = mid + 1;
        }
        return begin;
    }


你可能感兴趣的:(算法)