Python Leetcode(665.非递减数列)

Python 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]。

Solution:(设置一个flag统计数组中递减的次数，大于1次则不能成为非递减数列；然后判断i和i+2元素以及i-1和i+1元素的大小，不满足非递减则返回False)

class Solution(object):
    def checkPossibility(self, nums):
        """
        :type nums: List[int]
        :rtype: 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 in [0, len(nums)-2] or nums[i] <= nums[i+2] or nums[i-1] <= nums[i+1]:
                continue
            return False
        return True

solution = Solution()
print(solution.checkPossibility([2, 4, 2, 3]))

True