符合下列属性的数组 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 。
示例 :
输入:arr = [0,1,0]
输出:1
一看只要数据有排序规律的,就要想到二分法。
class Solution { public int peakIndexInMountainArray(int[] arr) { int left = 0; int right = arr.length-1; while (left<=right){ int mid = (left+right)/2;
//当left=0,right=1时,mid=0,此时应该将mid==1,不然下一步计算会数组越界。 if(mid==0){ mid ++; } if(arr[mid]>arr[mid-1] && arr[mid]>arr[mid+1]){ return mid; }else if(arr[mid] < arr[mid-1]){ right = mid-1; }else { left = mid+1; } } return left; } }
时间复杂度:O(log(n))
官方代码:我上面的代码还是把问题搞的更复杂了,其实只要找到 a[i]<a[i+1]最大值即可。
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; } }