1. 程式人生 > 其它 >力扣852(java&python3)-山脈陣列的峰頂索引(中等)

力扣852(java&python3)-山脈陣列的峰頂索引(中等)

題目:

符合下列屬性的陣列 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 是一個山脈陣列

進階:很容易想到時間複雜度 O(n) 的解決方案,你可以設計一個 O(log(n)) 的解決方案嗎?

來源:力扣(LeetCode)
連結:https://leetcode.cn/problems/peak-index-in-a-mountain-array
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。

解題思路:

【二分查詢】

其實看到提示中進階要求設計時間複雜度為O(log(n))的時候就會想到二分查詢,加上題目中給定的陣列可以看做兩個有序的陣列,先升序(上山)後降序(下山),目的就是找到升序中最後的那一個數字(山頂)的下標。用二分查詢:

  • 初始化左右邊界:left = 0, right = n - 1,計算出mid
  • 迴圈條件是:left < right,判斷的條件就讓mid與 mid - 1的數值進行判斷
    • arr[mid] > = arr[mid  - 1]:說明mid還處於上山階段,此時mid有可能就是山頂,需要往右繼續找山頂,搜尋區間變為[mid, right],即讓left = mid(注意:mid需要向上取整);
    • arr[mid] < arr[mid-1]:說明mid已經處於下山階段,就需要往左尋找山頂,搜尋區間變為[left, mid - 1],即讓right = mid - 1;
  • 迴圈結束的條件是:left == right,題目中說了 arr 是一個山脈陣列,那麼最後一定會是山頂,直接返回 left 或者 right 即可。

java程式碼(left < right):

 1 class Solution {
 2     public int peakIndexInMountainArray(int[] arr) {
 3         int n = arr.length;
 4         int left = 0, right = n - 1;
 5         while (left < right){
 6             int mid = left + (right - left + 1) / 2;
 7             if (arr[mid] > arr[mid -1]){
 8                 left = mid;
 9             }else{
10                 right = mid - 1;
11             }
12         }
13         return left;
14     }
15 }

 python3程式碼(left <= right):錯誤程式碼:

 1 class Solution:
 2     def peakIndexInMountainArray(self, arr: List[int]) -> int:
 3         left, right = 0, len(arr) - 1
 4         while left <= right:
 5             mid = left + (right - left) // 2
 6             if arr[mid] > arr[mid - 1] and arr[mid] > arr[mid + 1]:
 7                 return mid
 8             # arr[mid] > arr[mid - 1]並且arr[mid] <= arr[mid + 1]
 9             # mid + 1更大
10             elif arr[mid] > arr[mid - 1]:
11                 left = mid + 1
12             else:
13                 # arr[mid] > arr[mid + 1]並且arr[mid]<= arr[mid -1]
14                 # mid - 1更大
15                 right = mid - 1
16         return -1

有一個測試用例過不去:

 研究一下它為什麼過不去:

[3,9,8,6,4]

①left = 0, right = 4, mid = 2, arr[2] > arr[3]且arr[2] < arr[1]故right = mid -1 = 1;

②left = 0, right = 1,mid = 0, 這時mid -1 = -1已經陣列越界了,故為了避免這種判斷,arr[mid+1] > arr[mid] 必須寫在 arr[mid-1]>arr[mid] 前面判斷。跟著正確的,arr[1] > arr[0],left = mid + 1 = 1;

③left = 1,right = 1,mid = 1,arr[0] < arr[1] < arr[2],故直接返回mid = 1。

python3正確程式碼:

 1 class Solution:
 2     def peakIndexInMountainArray(self, arr: List[int]) -> int:
 3         left, right = 0, len(arr) - 1
 4         while left <= right:
 5             mid = left + (right - left) // 2
 6             if arr[mid] > arr[mid + 1] and arr[mid] > arr[mid - 1]:
 7                 return mid
 8              # arr[mid] > arr[mid + 1]並且arr[mid]<= arr[mid -1]
 9              # mid - 1更大
10             elif arr[mid] > arr[mid + 1]:
11                 right = mid - 1
12             else:
13                 # arr[mid] > arr[mid - 1]並且arr[mid] <= arr[mid + 1]
14                 # mid + 1更大
15                 left = mid + 1
16         return -1