【LeetCode】665. 非递减数列

题目描述

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

我们是这样定义一个非递减数列的： 对于数组中所有的 i (1 <= i < n)，满足 array[i] <= array[i + 1]。

示例 1:

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

示例 2:

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

说明:  n 的范围为 [1, 10,000]。

 

思路

观察 1,2,8,5,6 和 1,5,8,2,6 两个序列：当遇到第一次 nums[i] > nums[i+1] 时，需要判断 nums[i+1] 和 nums[i-1] 的大小，若前者大于后者，则把 nums[i] 改为 nums[i-1] 或 nums[i+1] 均可，序列变为 1,2,2,5,6 或 1,2,5,5,6；反之，nums[i+1] 应变为 nums[i]，即序列变为 1,5,8,8,6。此后，若还出现 nums[i] > nums[i+1]，则该数组不满足题意。

 

代码

class Solution {
public:
    bool checkPossibility(vector<int>& nums) {
        if(nums.size() == 0) return 0;
        if(nums.size() == 1) return true;
        int len = nums.size();
        int flag = 0;    //标记出现错误的次数
        for(int i=0; i<len-1; i++){
            if(nums[i] > nums[i+1]){
                if(flag >= 1){
                    return false;
                }
                flag++;
                if(nums[i+1] < nums[i-1]){
                    nums[i+1] = nums[i];
                }
                //else对后续判断也没有影响
            }
        }
        return true;
    }
};