1. 程式人生 > >Leetcode 162.尋找峰值

Leetcode 162.尋找峰值

尋找峰值

峰值元素是指其值大於左右相鄰值的元素。

給定一個輸入陣列 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 }