LeetCode Find Peak Element

来源:互联网 发布:python readline utf8 编辑:程序博客网 时间:2024/05/16 17:42

A peak element is an element that is greater than its neighbors.

Given an input array where num[i] ≠ num[i+1], find a peak element and return its index.

The array may contain multiple peaks, in that case return the index to any one of the peaks is fine.

You may imagine that num[-1] = num[n] = -∞.

For example, in array [1, 2, 3, 1], 3 is a peak element and your function should return the index number 2.

思路分析:这题O(N)的思路很容易想到,直接遍历找local max即可。但是还可以用二分的思路得到O(logN)的算法,具体而言,每次更新l或者r的标准是要保证剩下的半边一定要包含至少一个peak element。要做到这一点,我们只需要比较num[mid]和num[mid+1], 如果中间元素大于其相邻后续元素,则中间元素左侧(包含该中间元素)必包含一个局部最大值。如果中间元素小于其相邻后续元素,则中间元素右侧(不含该中间元素)必包含一个局部最大值。对于更新l和r的时候是否要包括中间元素要想清楚,考虑中间元素是否可能是解。

AC Code

public class Solution {    public int findPeakElement(int[] num) {        //0535        int l = 0;         int r = num.length - 1;        while(l <= r){            if(l == r) return l;            int mid = (l + r) / 2;            if(num[mid] > num[mid + 1]){                //find on the left                r = mid;            } else {                //find on the right                l = mid + 1;            }        }        return l;    }}


0 0