Lintcode阶梯训练~算法

寻找峰值

2017-06-21  本文已影响8人  lyoungzzz

题目

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

相邻位置的数字是不同的
A[0] < A[1] 并且 A[n - 2] > A[n - 1]
假定 P 是峰值的位置则满足A[P] > A[P-1]且A[P] > A[P+1],返回数组中任意一个峰值的位置。

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

样例

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

代码实现

class Solution {
    /**
     * @param A: An integers array.
     * @return: return any of peek positions.
     */
    public int findPeak(int[] A) {
        // write your code here
        int start = 0;
        int end = A.length - 1;
        while (start + 1 < end) {
            int mid = start + (end - start) / 2;
            if (A[mid - 1] <= A[mid] && A[mid] <= A[mid + 1]) {
                start = mid;
                continue;
            }
            if (A[mid - 1] >= A[mid] && A[mid] >= A[mid + 1]) {
                end = mid;
                continue;
            }
            if (A[mid - 1] <= A[mid] && A[mid] >= A[mid + 1]) {
                return mid;
            }
            start = mid;
        }
        if (A[start] < A[end]) {
            return end;
        } else {
            return start;
        }
    }
}

上一篇 下一篇

猜你喜欢

热点阅读