二分查找——山峰数组系列（leetcode 852， 剑指 Offer II 069）

"""
leetcode 852. 山脉数组的峰顶索引  剑指 Offer II 069. 山峰数组的顶部
符合下列属性的数组 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 是一个山脉数组

解题思路：
山脉数组也就是先递增后递减的数组，这道题可以转换成求解数组中最大值的索引,需要考虑两种情况：
1）数组全部升序；2）数组全部倒序；当数组全部降序的时候mid=0;当数组全部升序时，mid = len(arr) - 1
"""
class Solution(object):
    def peakIndexInMountainArray(self, arr):
        left = 0
        right = len(arr)
        while left < right:
            mid = left + int((right - left) / 2)
            # 针对越界情况进行判断（全部是递增 或者 全部递减）
            if mid + 1 >= len(arr) or mid - 1 < 0:
                return mid
            if arr[mid] > arr[mid + 1] and arr[mid] > arr[mid - 1]:
                return mid
            if arr[mid] > arr[mid - 1]:
                left = mid + 1
            if arr[mid] > arr[mid + 1]:
                right = mid

if __name__ == "__main__":
    arr = [0,2,1,0]
    solution = Solution()
    res = solution.peakIndexInMountainArray2(arr)
    print(res)