力扣 665. 非递减数列

题目

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

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

示例

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/non-decreasing-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：两次单调栈

Java实现

class Solution {
    public boolean checkPossibility(int[] nums) {
        int n = nums.length;
        Stack<Integer> stack1 = new Stack<>();
        Stack<Integer> stack2 = new Stack<>();
        int cnt1 = 0, cnt2 = 0;

        for (int i = 0; i < n; i++) {
            while (!stack1.isEmpty() && stack1.peek() > nums[i]) {
                stack1.pop();
                cnt1++;
            }
            stack1.push(nums[i]);
        }

        for (int i = n - 1; i >= 0; i--) {
            while (!stack2.isEmpty() && stack2.peek() < nums[i]) {
                stack2.pop();
                cnt2++;
            }
            stack2.push(nums[i]);
        }

        return cnt1 <= 1 || cnt2 <= 1;
    }
}

方法2：数组

1、非递减数列的概念
a 递增数列：【1，2，3，4，6，8】
b【1，2，2，2，6】

2、递减数列
如：【4，3，2，1】，flag记录 nums[i]>nums[i+1] (当前项比后一项小个数)
flag>1的话直接返回false（因为只能修改一个数）

3、如果出现【3，4，1，2】的情况：(i>0)
nums[i] > nums[i+1]，并且 nums[i-1] > nums[i+1]，就让nums[i+1] = nums[i]

Python实现

class Solution:
    def checkPossibility(self, nums: List[int]) -> bool:
        n = len(nums)
        flag=0
        for i in range(n-1):
            if nums[i] > nums[i+1]:
                flag+=1
                if flag>1:
                    return False
                if i>0 and nums[i+1]<nums[i-1]:
                    nums[i+1]=nums[i]
        return True

Java实现

class Solution {
    public boolean checkPossibility(int[] nums) {
        // 如果数组长度小于2，必可以纠正
        if (nums.length <= 2) {
            return true;
        }
        int count = 0;
        for (int i = 1; i < nums.length - 1; i++) {
            if (nums[i - 1] > nums[i]) {
                if (i > 1 && nums[i - 2] > nums[i] && nums[i - 1] > nums[i + 1]) {
                    return false;
                } else {
                    count++;
                }
            }
        }
        if (nums[nums.length - 2] > nums[nums.length - 1]) {
            count++;
        }
        if (count < 2) {
            return true;
        } else {
            return false;
        }
    }
}