数据结构与算法之数组寻找峰值分而治之

news2024/12/30 2:44:08

这一篇分享一道在数组中寻找峰值的题目，使用到分而治之，二分查找等思想。本文章会讲解这个二分查找的本质，以及为什么要用二分查找，它能解决哪一类题目？

目录：
一.题目及其要求
1.分而治之
2.题目思路
3.具体做法+配图演示
4.复杂度分析
二.剖析二分查找
1.本质
2.优点
3.适用场景

一.题目及其要求

1.给定一个长度为n的数组，返回其中任何一个峰值的索引
2.峰值元素是指其值严格大于左右相邻值的元素
3.数组两个边界可以看成是最小， nums[−1]= nums[ n]=−∞
4.峰值不存在平的情况，即相邻元素不会相等
5.峰值从第二个数开始，不考虑边界的-∞

如输入[2,4,1,2,7,8,4]时，会形成两个山峰，一个是索引为1，峰值为4的山峰，另一个是索引为5，峰值为8的山峰，如下图所示：

示例：

输入：[2,4,1,2,7,8,4]
返回值：1 说明：4和8都是峰值元素，返回4的索引1或者8的索引5都可以

输入：[1,2,3,1]
返回值：2 说明：3 是峰值元素，返回其索引 2

1.分而治之

分治即“分而治之”，意思就是把一个复杂大问题分成若干个简单的小问题，当小问题解决后，大问题也就迎刃而解。“分”指的是将一个大而复杂的问题划分成多个性质相同但是规模更小的子问题，子题继续按照这样划分，直到问题可以被轻易解决；“治”指的是将子问题单独进行处理。经过分治后的子问题，将若干个子问题进行合并才能得到原问题的解，因此整个分治过程经常用递归来实现。

2.题目思路

因为题目只需要找到其中一个波峰，因此不断地往高处走，一定会有波峰。那我们可以每次找一个标杆元素，将数组分成两个区间，每次就较高的一边走，因此也可以用分治来解决，而标杆元素可以选择区间中点。

//右边是往下，不一定有坡峰
if(nums[mid] > nums[mid + 1])
    right = mid;
//右边是往上，一定能找到波峰
else
    left = mid + 1;

下面配图进行理解上面代码：

3.具体做法+配图演示

1：二分查找首先从数组首尾开始，每次取中间值，直到首尾相遇。

2：如果中间值的元素大于它右边的元素，说明往右是向下，我们不一定会遇到波峰，但是那就往左收缩区间。

3：如果中间值大于右边的元素，说明此时往右是向上，向上一定能有波峰，那我们往右收缩区间。

4：最后区间收尾相遇的点一定就是波峰。

下面是配图演示：

4.代码实现

class Solution {
public:
    int findPeakElement(vector<int>& nums) {
        int left = 0;
        int right = nums.size() - 1;
        //二分法
        while(left < right){ 
            int mid = (left + right) / 2;
            //右边是往下，不一定有坡峰
            if(nums[mid] > nums[mid + 1])
                right = mid;
            //右边是往上，一定能找到波峰
            else
                left = mid + 1;
        }
        //其中一个波峰
        return right; 
    }
};