粤嵌—2024/3/22—寻找峰值

嵌入式小小聪

已于 2024-04-15 19:50:01 修改

阅读量237

点赞数 1

分类专栏：粤嵌每日一题文章标签： leetcode

于 2024-04-15 19:47:35 首次发布

本文链接：https://blog.csdn.net/qq_68915288/article/details/137792698

版权

粤嵌每日一题专栏收录该内容

52 篇文章 0 订阅

订阅专栏

本文介绍了两种方法寻找数组中的峰值元素：一种是线性遍历，适用于小型数组；另一种是二分法，通过不断缩小区间判断峰值位置，效率更高。着重讨论了单调性和相邻数不相等的假设条件。

摘要由CSDN通过智能技术生成

代码实现：

方法一：遍历
int findPeakElement(int *nums, int numsSize) {
    if (numsSize == 1 || nums[0] > nums[1]) {
        return 0;
    }
    if (nums[numsSize - 1] > nums[numsSize - 2]) {
        return numsSize - 1;
    }
    int i;
    for (i = 1; i < numsSize; i++) {
        if (nums[i] > nums[i - 1] && nums[i] > nums[i + 1]) {
            break;
        }
    }
    return i;
}
方法二：二分法

数组的两端nums[-1] = nums[n] = -∞ 都是负无穷，因此数组无论是单调递增还是单调递减，又或者是成起伏状，数组中必定包含一个峰值。如下图所示：

因为数组中的峰值不止一个，我们找到任意一个即可。题目还告诉我们对于所有有效的i都有 nums[i] != nums[i + 1]，即数组中的任意两个相邻数都不相等

使用二分查找，每次找出区间的中点mid，比较nums[mid]与nums[mid + 1]的大小关系来推断哪个区间内一定存在峰值，然后取一定存在峰值的区间。这样不断缩小区间范围，区间所剩下的最后一个数就是答案

如果nums[mid] > nums[mid + 1]，那么在[l, mid]这个区间内一定存在一个峰值

如果nums[mid] < nums[mid + 1]，那么在[mid+1, r]这个区间内一定存在一个峰值
int findPeakElement(int *nums, int numsSize) {
    int l = 0, r = numsSize - 1, mid; // 左闭右闭
    while (l <= r) {
        mid = (l + r) >> 1;
        if ((mid == 0 || nums[mid] > nums[mid - 1]) && (mid == numsSize - 1 || nums[mid] > nums[mid + 1])) { // 找到了峰值
            break;
        }
        if (nums[mid] > nums[mid + 1]) { // [l, mid - 1]这个区间一定存在一个峰值
            r = mid - 1;
        } else { // [mid + 1, tail]这个区间一定存在一个峰值
            l = mid + 1;
        }
    }
    return mid;
}