Leetcode之二分法专题-852. 山脉数组的峰顶索引（Peak Index in a Mountain Array）

Leetcode之二分法专题-852. 山脉数组的峰顶索引（Peak Index in a Mountain Array）

---

 

我们把符合下列属性的数组 A 称作山脉：

• A.length >= 3
• 存在 0 < i < A.length - 1 使得A[0] < A[1] < ... A[i-1] < A[i] > A[i+1] > ... > A[A.length - 1]

给定一个确定为山脉的数组，返回任何满足 A[0] < A[1] < ... A[i-1] < A[i] > A[i+1] > ... > A[A.length - 1] 的 i 的值。

 

示例 1：

输入：[0,1,0]
输出：1

示例 2：

输入：[0,2,1,0]
输出：1

 

提示：

1. 3 <= A.length <= 10000
2. 0 <= A[i] <= 10^6
3. A 是如上定义的山脉

---

 

分析：二分法，题中要求返回一个符合要求的数组下标。

利用二分法，首先一分为二，然后取mid，判断是否符合 A[i-1] < A[i] > A[i+1]

如果不符合，则又分为两种情况：

　　1、A[i-1] < A[i] < A[i+1] 这种情况则把left置为mid+1

　　2、A[i-1] > A[i] > A[i+1] 这种情况把right置为mid-1

如果符合，则直接返回下标。

 

AC代码：

class Solution {
    public int peakIndexInMountainArray(int[] A) {
        if(A==null || A.length==0){
            return 0;
        }
        int left = 0;
        int right = A.length-1;
        while(left<=right){
            int mid = (left+right)/2;
            if(A[mid-1]<A[mid] && A[mid]>A[mid+1]){
                return mid;
            }else if(A[mid-1]>A[mid]){
                right = mid-1;
            }else {
                left = mid+1;
            }
        }
        
        return 0;
    }
}