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

思路

经过多个样例的观察，得出一个规律：逆序的那组数据，将其中一个改成另一个，如果能够使整个序列都是升序，则返回TRUE,否则返回FALSE。我的代码如下：

class Solution:
    def checkPossibility(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        lens = len(nums)
        flag = 0 #用于判断是否有逆序
        for i in range(lens-1):
            if nums[i]>nums[i+1]:
                flag = 1
                x = nums[i]
                nums[i] = nums[i+1] #将前面一个数改成后面一个数
                if nums == sorted(nums): #判断修改后的muns是否升序
                    return True
                else:
                    nums[i] = x
                    nums[i+1] = x #将后面一个数改成前面一个数
                    if nums == sorted(nums):#判断修改后的muns是否升序
                        return True
                    else:
                        return False
        if flag==0:
            return True

由于我的代码用时较长，分析原因如下，使用了两次sorted()函数

学习他人代码

class Solution:
    def checkPossibility(self, nums):
        """
        :type nums: List[int]
        :rtype: bool
        """
        if len(nums)<=2:return True #判断只有两种元素的情况，肯定是返回True
        n = 0 if nums[0] <= nums[1] else 1 #先判断nums[0]和nums[1]的情况，因为下面是比较nums[i-1] > nums[i],且涉及到i-2

        for i in range(2,len(nums)):
            if nums[i-1] > nums[i]:
                n += 1 
                if n > 1:return False #超过一次
                if nums[i-2] > nums[i]:
                    nums[i] = nums[i-1]
                else:
                    nums[i-1] = nums[i-2] 
        return True