力扣 leetcode 665. 非递减数列 (python)

Topic

给你一个长度为 n 的整数数组，
请你判断在 最多 改变 1 个元素的情况下，该数组能否变成一个非递减数列。
我们是这样定义一个非递减数列的： 对于数组中所有的 i (0 <= i <= n-2)，总满足 nums[i] <= nums[i + 1]。

Example_1

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

Example_2

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

Tips

1 <= n <= 10 ^ 4

• 10 ^ 5 <= nums[i] <= 10 ^ 5

Solution

我们可以利用类似贪心算法的思路进行判断

首先我们可以先进行简单的判断
如果列表中出现了两次递减
那么直接返回False

用count进行判断
一次递减count次数+ 1
count大于一后返回False

之后对只出现一次递减的情况进行判断
通过改变数组后进行判断

先判断特殊情况
如果出现的是第一位比第二位大
则将第一位变为第二位的值

正常情况：
可以有两种情况改变
第一种是改变后者较小的值为更大的值

当遍历的值后面的值比前一个值更大
则总体上除了遍历的值后面的值以外是递增的
则将遍历的值后面的值改成遍历的值

第二种是改变大的值为之前的值

当遍历的值后面的值比前一个值更小
则总体上除了遍历的值以外是递增的
则将遍历的值改成与遍历值后面的相等

Code

class Solution:
    def checkPossibility(self, nums: List[int]) -> bool:
        count = 0

        for i in range(len(nums) - 1):
            if nums[i] > nums[i + 1]:
                count += 1
                if count > 1:
                    return False
                if i == 0:
                    nums[i] = nums[i + 1]
                else:
                    if nums[i - 1] > nums[i + 1]:
                        nums[i + 1] = nums[i] 
                    else:
                        nums[i] = nums[i + 1]
                        
        return True

Result