LeetCode-162-尋找峰值

雄獅虎豹發表於2022-02-02
LeetCode

尋找峰值

題目描述：峰值元素是指其值大於左右相鄰值的元素。

給你一個輸入陣列 nums，找到峰值元素並返回其索引。陣列可能包含多個峰值，在這種情況下，返回 任何一個峰值 所在位置即可。

你可以假設 nums[-1] = nums[n] = -∞ 。

示例說明請見LeetCode官網。

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/probl...
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

解法一：陣列遍歷

首先，判斷兩種特殊情況：

  • 如果只有一個元素，直接返回第一個元素的索引位；
  • 如果有兩個元素，則根據判斷這兩個元素的大小直接返回相應的索引位。

如果陣列nums不只有2個元素，則從第一個元素開始遍歷判斷是否是峰值：

  • 首先判斷陣列的第一位，如果大於第二位的值，則返回0；
  • 否則，遍歷從1到nums.length-2的元素，判斷是否是峰值，如果是峰值，直接返回相應的索引位；
  • 最後，如果沒有找到峰值，判斷最後一個元素是否大於前面的一個元素值，如果是，則返回最大的索引位，否則，返回-1表示沒有峰值。
public class LeetCode_162 {
    public static int findPeakElement(int[] nums) {
        // 如果只有一個元素，直接返回第一個元素的索引位
        if (nums.length == 1) {
            return 0;
        }
        // 如果有兩個元素，則根據判斷這兩個元素的大小直接返回相應的索引位
        if (nums.length == 2) {
            if (nums[0] > nums[1]) {
                return 0;
            } else {
                return 1;
            }
        }
        // 特殊判斷陣列的第一位，如果大於第二位的值，則返回0
        if (nums[0] > nums[1]) {
            return 0;
        }
        int index = 1;
        // 遍歷從1到nums.length-2的元素，判斷是否是峰值
        for (; index < nums.length - 1; index++) {
            if (nums[index] > nums[index - 1]) {
                if (nums[index] > nums[index + 1]) {
                    return index;
                }
            } else {
                if (nums[index] > nums[index + 1]) {
                    index++;
                }
            }
        }
        // 如果前面沒有找到峰值，判斷最後一個元素是否大於前面的一個元素值，如果是，則返回最大的索引位，否則，返回-1表示沒有峰值
        if (nums[nums.length - 1] > nums[nums.length - 2]) {
            return nums.length - 1;
        } else {
            return -1;
        }
    }

    public static void main(String[] args) {
        int[] nums = new int[]{1, 2, 1, 3, 5, 6, 4};
        // 測試用例，期望返回值： 1或5
        System.out.println(findPeakElement(nums));
    }
}
【每日寄語】 你想賺這份錢，你就得吃這份苦。要做生活的強者，就不能在乎得失。

相關文章