《剑指 Offer》专项突破版 - 面试题 68 : 查找插入位置/ 69 : 山峰数组的顶部（C++ 实现）

目录

面试题 68 : 查找插入位置

面试题 69 : 山峰数组的顶部

---

 

---

面试题 68 : 查找插入位置

题目：

输入一个排序的整数数组 nums 和一个目标指 t，如果数组 nums 中包含 t，则返回 t 在数组中的下标；如果数组 nums 中不包含 t，则返回将 t 按顺序插入数组 nums 中的下标。假设数组中没有相同的数字。例如，输入数组 nums 为 [1, 3, 6, 8]，如果目标值 t 为 3，则输出 1；如果 t 为 5，则返回 2。

分析：

首先考虑如果目标值 t 不在数组中时它应该被插入哪个位置。由于数组是排序的，因此它应该排在所有比它小的数字的后面。也就是说，它的插入位置满足两个条件：一是该位置上的数字大于 t，二是该位置的前一个数字小于 t（总结下来就是：数组中第一个大于 t 的数字的位置）。例如，当数组为 [1, 3, 6, 8]，目标值为 5，它将被插入下标为 2 的位置，该位置当前的值为 6，大于目标值，该位置的前一个值是 3，小于目标值。

当数组中包含目标值时，返回它在数组中的位置。

综上所述，即查找数组中第一个大于或等于 t 的数字的位置。

代码实现：

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;
        while (left <= right)
        {
            int mid = (left + right) / 2;
            if (nums[mid] < target)
                left = mid + 1;
            else  // nums[mid] >= target
                right = mid - 1;
        }
        return left;
    }
};

当 while 循环结束（left > right），left 左边的数字都小于 target，right 右边的数字都大于或等于 target，因此 right + 1，即 left 就是第一个大于或等于 target 的数字的位置。

---

面试题 69 : 山峰数组的顶部

题目：

符合下列属性的数组 arr 称为山峰数组：

• arr.length >= 3

• 存在 i（0 < i < arr.length - 1） 使得：arr[0] < arr[1] < ··· < arr[i - 1] < arr[i] > arr[i + 1] > ··· > arr[arr.length - 1]

给定由整数组成的山峰数组 arr，返回下标 i，即山峰顶部（数组中最大值的位置）。

例如，在数组 [1, 3, 5, 4, 2] 中，最大值是 5，输出它在数组中的下标 2。

分析：

不难想到直观的解法来解决这个题目，即逐一扫描整个数组，通过比较就能找出数组中的最大值。显然，这种解法的时间复杂度是 O(n)。这种解法对任意数组都适用，并没有充分利用这个题目的特点，即数组先递增再递减。由于问题是关于在排序数组中查找数字，虽然整个数组并不是排序的，但分成前后两段后每段都分别排序，因此二分查找算法值得一试。

山峰数组中的最大值是数组中唯一一个比它左右两边数字都大的数字。位于最大值前面的数字（除第 1 个数字之外）总是比它前一个数字大但比它后一个数字小；位于最大值后面的数字（除最后一个数字之外）总是比它前一个数字小但比它后一个数字大。

可以根据山峰数组的这个特点应用二分查找算法。先取出数组中间的数字。

1. 如果这个数字比它前后两个数字都大，那么就找到了数组的最大值。

2. 如果这个数字比它前一个数字大但比后一个数字小，那么这个数字位于数组递增的部分，数组的最大值一定在它的后面，接下来只需要在数组的后半部分查找就可以。

3. 如果这个数字比它前一个数字小但比后一个数字大，那么这个数字位于数组递减的部分，数组的最大值一定在它的前面，接下来只需要在数组的前半部分查找就可以。

代码实现：

class Solution {
public:
    int peakIndexInMountainArray(vector<int>& arr) {
        int left = 1;
        int right = arr.size() - 2;
        while (left <= right)
        {
            int mid = (left + right) / 2;
            if (arr[mid] > arr[mid - 1] && arr[mid] > arr[mid + 1])
                return mid;
            
            if (arr[mid] > arr[mid - 1])
                left = mid + 1;
            else
                right = mid - 1;
        }
        return -1;
    }
};

注意：

1. 在一个长度为 n 的山峰数组中，由于第 1 个数字和最后一个数字都不可能是最大值，因此初始查找范围为数组下标从 1 到 n - 2 的部分。

2. 如果输入的数组是一个有效的山峰数组，那么 while 循环中一定能找到山峰数组的最大值。只是 C++ 的语法要求函数的每个分支必须有返回值，所以在函数体的最后添加一行返回 -1 的代码。实际上，这一行代码不会被执行。