BM19 寻找峰值

题目描述

给定一个长度为n的数组nums，请你找到峰值并返回其索引。数组可能包含多个峰值，在这种情况下，返回任何一个所在位置即可。
1.峰值元素是指其值严格大于左右相邻值的元素。严格大于即不能有等于
2.假设 nums[-1] = nums[n] = -\infty−∞
3.对于所有有效的 i 都有 nums[i] != nums[i + 1]
4.你可以使用O(logN)的时间复杂度实现此问题吗？

解题思路

用二分法求解峰值问题。对于每一个位置，判断其两边的元素于他的大小。在比他大的那一侧一定存在峰值。

代码

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @return int整型
     */
    int findPeakElement(vector<int>& nums) {
        // write code here
        int m = INT_MIN;
        int l = nums.size();
        if(l == 1)
            return 0;
        
        int ll = 0;
        int rr = l;
        while(ll <= rr)
        {
            int mid = (ll + rr) >> 1;
            if(mid!=0 && mid!=l - 1)
            {
                if(nums[mid] > nums[mid - 1] && nums[mid] > nums[mid + 1])
                    return mid;
                else if(nums[mid] < nums[mid - 1])
                {
                    rr = mid - 1;
                }
                else
                    ll = mid + 1;
            }
            else if(mid == 0)
            {
                if(nums[mid] > nums[mid + 1])
                    return mid;
                ll = mid + 1;
            }
            else
            {
                if(nums[mid] > nums[mid - 1])
                    return mid;
                rr = mid - 1;
            }
        }
        return -1;
        
    }
};