Leetcode 665. 非递减数列

本题链接

题目

给你一个长度为 n 的整数数组，请你判断在 最多 改变 1 个元素的情况下，该数组能否变成一个非递减数列。

我们是这样定义一个非递减数列的： 对于数组中任意的 i (0 <= i <= n-2)，总满足 nums[i] <= nums[i + 1]。

示例 1:

输入: nums = [4,2,3]
输出: true
解释: 你可以通过把第一个4变成1来使得它成为一个非递减数列。

示例 2:

输入: nums = [4,2,1]
输出: false
解释: 你不能在只改变一个元素的情况下将其变为非递减数列。

提示：

1 <= n <= 10 ^ 4
- 10 ^ 5 <= nums[i] <= 10 ^ 5

---

思路

先遍历数组，如果出现 nums[i] > nums[i + 1] 的情况就说明原数组出现递减的情况，要改成非递减数列的话可以修改 nums[i] 或者 nums[i + 1]。

如果修改 nums[i]，由于一定有 nums[i - 1] <= nums[i + 1]，所以直接把 nums[i] 改成 nums[i + 1] 即可，然后判断修改后，从 nums[i + 1] 开始是否是非递减的，如果是，则成功，如果不是，那么继续修改 nums[i + 1]。

如果修改 nums[i + 1]，一方面要保证修改后的 nums[i + 1] 满足 nums[i] <= nums[i + 1]，另一方面修改后的 nums[i + 1] 应该尽量小，防止出现 nums[i + 1] > nums[i + 2] 的情况。所以将 nums[i + 1] 改成 nums[i] 即可。随后检查从 nums[i + 1] 开始是否是非递减的，如果是，则成功，如果不是，失败。

时间复杂度：$O(n)$
空间复杂度：$O(1)$

C++ 代码

class Solution {
public:
    bool checkPossibility(vector<int>& nums) {
        int size = nums.size();
        for (int i = 0; i < size - 1; ++i) {
            if (nums[i] > nums[i + 1]) {
                // 先尝试修改nums[i]，检查剩下的是否符合要求
                int temp = nums[i];
                nums[i] = nums[i + 1];
                if (is_sorted(nums.begin(), nums.end()))
                    return true;
                nums[i] = temp;     // 复原数据

                // 再尝试修改nums[i+1]，检查剩下的是否符合要求
                nums[i + 1] = nums[i];
                return is_sorted(nums.begin(), nums.end());
            }
        }
        return true;    // 没遇到问题，本身就符合要求
    }
};