leetcode852

山脉数组的峰顶索引
Category Difficulty Likes Dislikes
algorithms Easy (71.36%) 313 -
Tags
Companies
符合下列属性的数组 arr 称为 山脉数组 ：
arr.length >= 3
存在 i（0 < i < arr.length - 1）使得：
arr[0] < arr[1] < ... arr[i-1] < arr[i]
arr[i] > arr[i+1] > ... > arr[arr.length - 1]
给你由整数组成的山脉数组 arr ，返回任何满足 arr[0] < arr[1] < ... arr[i - 1] < arr[i] > arr[i + 1] > ... > arr[arr.length - 1] 的下标 i 。

示例 1：

输入：arr = [0,1,0]
输出：1
示例 2：

输入：arr = [0,2,1,0]
输出：1
示例 3：

输入：arr = [0,10,5,2]
输出：1
示例 4：

输入：arr = [3,4,5,1]
输出：2
示例 5：

输入：arr = [24,69,100,99,79,78,67,36,26,19]
输出：2

提示：

3 <= arr.length <= 104
0 <= arr[i] <= 106
题目数据保证 arr 是一个山脉数组

进阶：很容易想到时间复杂度 O(n) 的解决方案，你可以设计一个 O(log(n)) 的解决方案吗？

Discussion | Solution

class Solution {
public:
    /**
      本质还是二分，可以将数组分为两个或三个部分
      前半部分是爬山。后半部分是下山。
      找到山顶返回。
      很多时候如果可以确定一个点是不是就和好写
      如果无法确定一个点是不是最终结果，只能不断划分区间就需要额外考虑区间的问题
    */
    int peakIndexInMountainArray(vector<int>& arr) {
        int l = 1, r = arr.size() - 2;
        while(l <= r)
        {
            int mid = l + ((r-l)>>1);
            if(arr[mid] > arr[mid-1] && arr[mid] > arr[mid+1])
                return mid;
            else if(arr[mid] > arr[mid - 1] && arr[mid] < arr[mid + 1])
                l = mid + 1;
            else 
                r = mid - 1;
        }
        return -1;
    }
};