Leetcode 162.尋找峰值
阿新 • • 發佈:2018-12-31
尋找峰值
峰值元素是指其值大於左右相鄰值的元素。
給定一個輸入陣列 nums,其中 nums[i] ≠ nums[i+1],找到峰值元素並返回其索引。
陣列可能包含多個峰值,在這種情況下,返回任何一個峰值所在位置即可。
你可以假設 nums[-1] = nums[n] = -∞。
示例 1:
輸入: nums = [1,2,3,1]
輸出: 2
解釋: 3 是峰值元素,你的函式應該返回其索引 2。
示例 2:
輸入: nums = [1,2,1,3,5,6,4]
輸出: 1 或 5
解釋: 你的函式可以返回索引 1,其峰值元素為 2;
或者返回索引 5, 其峰值元素為 6。
說明:
你的解法應該是 O(logN) 時間複雜度的。
題目說了相鄰元素不會相等,這個條件很重要。
nums[mid] < nums[mid + 1],
說明mid與後一個位置形成遞增區間,則mid後面一定存在峰且當前mid一定不是峰,
則low右移動mid一個位置(這個位置就有可能是峰了)
nums[mid] > nums[mid + 1],
說明mid與後一個位置形成遞減區間,則當前位置mid就有可能是峰(也可能在其前面),
則high左移動到mid
當low和high相等時是否能得到結果了?即是否應該截止?
因為high的後面是下降的,而low的前面是上升的,所以相等時,已經可以得到結果。
1 public class Solution{ 2 public static int findPeakElement(int[] nums) { 3 int low=0,high=nums.length-1; 4 while(low<high){ 5 int mid=(low+high)/2; 6 if(nums[mid]<nums[mid+1]){ 7 low=mid+1;8 }else{ 9 high=mid; 10 } 11 } 12 return low; 13 } 14 15 public static void main(String[] args){ 16 int[] nums={1,3,2,4,5,6,7}; 17 findPeakElement(nums); 18 } 19 }