- 掘金团队号上线,助你 Offer 临门! 点击 查看活动详情
本题核心
每天一道简单算法,防止老年痴呆~
题目描述
峰值元素是指其值大于左右相邻值的元素。
给你一个输入数组 nums,找到峰值元素并返回其索引。数组可能包含多个峰值,在这种情况下,返回 任何一个峰值 所在位置即可。
你可以假设 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。
提示:
1 <= nums.length <= 1000
-231 <= nums[i] <= 231 - 1
对于所有有效的 i 都有 nums[i] != nums[i + 1]
解题思路
一开始我老老实实的根据题目意思进行编写逻辑,判断数组i位的值一定要大于i-1和i+1。 看到题目后恍然大悟,如果i位的数字大于i+1位,那么i位就一定是峰值.
解题代码
老实人版本
public int findPeakElement(int[] nums) {
if(nums.length==1){
return 0;
}
for (int i = 0; i < nums.length; i++) {
if (i == 0) {
if (nums[i] > nums[i + 1]) {
return 0;
}
}
if (i == nums.length - 1) {
if (nums[i] > nums[i - 1]) {
return nums.length - 1;
}else{
break;
}
}
if(i-1>=0 && i+1<nums.length){
if (nums[i] > nums[i - 1] && nums[i] > nums[i + 1]) {
return i;
}
}
}
return 0;
}
官方正解
public int findPeakElement(int[] nums) {
for (int i = 0; i < nums.length-1; i++) {
if(nums[i]>nums[i+1]){
//左边的值不用做判断,因为在对比右边的值是就间接比对了左边
return i;
}
}
return nums.length-1;
}
##总结
简单算法,对于防止老年痴呆,非常好用👌