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的左右比较判断单调性，来二分

 

 

优秀无bug代码

O(n)

class Solution {
    public int peakIndexInMountainArray(int[] A) {
        int i = 0;
        while (A[i] < A[i+1]) i++;
        return i;
    }
}

O(logn)

class Solution {
    public int peakIndexInMountainArray(int[] A) {
        int lo = 0, hi = A.length - 1;
        while (lo < hi) {
            int mi = lo + (hi - lo) / 2;
            if (A[mi] < A[mi + 1])
                lo = mi + 1;
            else
                hi = mi;
        }
        return lo;
    }
}

 

下面的代码，题目说了长度大于等于3，不然gg，还有如果最值在边边也gg

public class 山脉数组的峰顶索引 {

    public static int peakIndexInMountainArray(int[] A) {
    	int i=0,j=A.length-1;
    	while(i<j) {
    		int mid = i + (j-i)/2;
    		if(A[mid] < A[mid+1])
    			i = mid;
    		else if(A[mid] < A[mid-1])
    			j = mid;
    		else
    			return mid;
    	}
        return i;
    }
    
	public static void main(String[] args) {
		int[] A = new int[] {0,1,3,2};
		System.out.println(peakIndexInMountainArray(A));
	}

}