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(n)。

第二种，二分查找：

• 首先题目给出，默认nums[-1] = nums[n] = -∞，所以峰值必定存在。
• 当nums[mid]>nums[mid+1]，即左边元素大于右边，走下坡路，那么左边必定有峰值，所以令right = mid。
• 同理nums[i]<=nums[i+1]，走上坡路，因此右边必定有峰值，所以令left = mid + 1，这里为什么要加1呢，是为了防止程序进入死循环。
• 由以上结论得二分查找

代码

public class problem162 {
	public int findPeakElement(int[] nums) {
		int left = 0;
		int right = nums.length-1;
		int mid;
		
		/*
		 * 利用二分查找
		 */
		while(left<right){
			mid = left + (right - left)/2;
			/*
			 * 说明当前为下坡，下坡峰值必在左侧
			 */
			if(nums[mid]>nums[mid+1])
				right = mid;
			else
				left = mid + 1;
		}
		
		return left;
    }
	public static void main(String[] args) {
		problem162 pro = new problem162();
		int[] nums = {1,2,3,1};
		System.out.println(pro.findPeakElement(nums));
	}
}