寻找峰值

问题描述：

峰值元素是指其值大于左右相邻值的元素。
给定一个输入数组 nums，其中 nums[i] ≠ nums[i+1]，找到峰值元素并返回其索引。
数组可能包含多个峰值，在这种情况下，返回任何一个峰值所在位置即可。
你可以假设 nums[-1] = nums[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。

说明:
你的解法应该是 O(logN) 时间复杂度的。

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

思路：

首先，得到数组大小n=nums.size();
因为数组的左右两边的值都为负无穷。如果数组中只有一个数，那么，唯一的数为峰值，如果，数组中有两个数，较大的数为峰值。
当数组的个数大于等于3时，第一种情况，在除两头的中间有峰值，此时峰值的位置满足nums[i] > nums[i - 1] && nums[i] > nums[i + 1]。
另外一种情况，如果中间没有峰值，那么峰值最出现在第一个位置或者最后一个位置。只需判断nums[0]是否大于nums[1]，或者 nums[n - 1]是否大于nums[n - 2]即可。

题解：

class Solution {
public:
    int findPeakElement(vector<int> &nums) {
        int n = nums.size();
        int flag = 0;
        int location;
        if(n==1){
            location=0;
        }
        else if(n==2){
            if(nums[0]>nums[1]){
                location=0;
            } else{
                location=1;
            }
        }
        else{
            for (int i = 1; i <= n - 2; i++) {
                if (nums[i] > nums[i - 1] && nums[i] > nums[i + 1]) {
                    flag = 1;
                    location = i;
                    break;
                }
            }
            if (flag == 0) {
                if (nums[0] > nums[1]) {
                    location = 0;
                }
                if (nums[n - 2] < nums[n - 1]) {
                    location = n - 1;
                }
            }
        }

        return location;
    }
};