力扣162-寻找峰值-C++两种解法

一、运行结果

顺序查找结果：

二分查找结果：

  

二、题目

峰值元素是指其值严格大于左右相邻值的元素。

给你一个整数数组 nums，找到峰值元素并返回其索引。数组可能包含多个峰值，在这种情况下，返回 任何一个峰值 所在位置即可。

你可以假设 nums[-1] = nums[n] = -∞ 。

你必须实现时间复杂度为 O(log n) 的算法来解决此问题。

示例 1：

输入：nums = [1,2,3,1]
输出：2
解释：3 是峰值元素，你的函数应该返回其索引 2。
示例 2：

输入：nums = [1,2,1,3,5,6,4]
输出：1 或 5 
解释：你的函数可以返回索引 1，其峰值元素为 2；
     或者返回索引 5， 其峰值元素为 6。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-peak-element

三、思路

查找峰值有两种方式，比较简单的是顺序查找，从前往后搜索，时间复杂度稍高，但数据量不大，效率也不低；另一种方法是二分搜索，时间复杂度为O(logn)。

四、代码

顺序查找，时间复杂度O(n):

class Solution {
public:
    int findPeakElement(vector<int>& nums) {
        int len = nums.size();
        if(len == 1 || nums[0] > nums[1]) return 0;
        for(int i = 1; i< len -1 ; i++){
            if(nums[i] > nums[i-1] && nums[i] > nums[i+1]){
                return i;
            }
        }
        return len-1;  //前面的都不符合
    }
};

二分查找，时间复杂度为O(n):

class Solution {
public:
    int findPeakElement(vector<int>& nums) {
        int len = nums.size();
        int left = 0, right = len - 1, mid;
        while(left < right){
            mid = left + (right - left) / 2;
            if(nums[mid] > nums[mid + 1]){
                right = mid;
            }else{
                left = mid + 1;
            }
        }
        return right;
    }
};