题目
给定一个长度为 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