1. 程式人生 > >leetcode 852. 山脈陣列的峰頂索引 【二分查詢】

leetcode 852. 山脈陣列的峰頂索引 【二分查詢】

題目:

我們把符合下列屬性的陣列 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 是如上定義的山脈

二分查詢(折半查詢):要求線性表是有序的,即表中記錄按關鍵字有序。對於有n個記錄的查詢表,進行二分查詢的時間複雜度為:log_{2}n

思路:

     陣列排序是有序的(不是嚴格意義上的遞增或者遞減序列,但是也是按照一定規則排列)。因此,這裡用二分查詢很合適。

程式碼:

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