给定一个长度为 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]。
思路:正确理解题目意思很重要,第一次看完觉得很简单,只要遍历一遍,判断一下相邻两个数后者是否小于前者,若是则计数变量加一,最后判断计数变量是否超过1即可。这个理解是错误的,至少是有漏洞,除了判断 nums[i-1]和nums[i]大小之外,还得比较nums[i-1]和nums[i+1],如果nums[i-1]大于nums[i] 且nums[i-1]大于 nums[i+1],则不可能完成;如果nums[i-1]大于nums[i] 但 nums[i-1]小于 nums[i+1],则只需让nums[i] = nums[i+1]即可。
本人方法:
class Solution:
def checkPossibility(self, nums: List[int]) -> bool:
if len(nums)<=2:
return True
else:
count=0 if nums[0] <= nums[1] else 1
n = len(nums)
for i in range(2,len(nums)):
if nums[i] < nums[i-1]:
count += 1
if count>1:
return False
if nums[i] < nums[i-2]:
nums[i] = nums[i-1]
else:
nums[i-1] = nums[i-2]
return count<=1
其他同学方法(思路其实一样):
class Solution:
def checkPossibility(self, nums):
"""
:type nums: List[int]
:rtype: bool
"""
count = 0
for i in range(1, len(nums)):
if nums[i] < nums[i-1] :
count += 1
if i - 2 < 0 or nums[i-2] < nums[i]:
nums[i-1] = nums[i]
else:
nums[i] = nums[i-1]
if count > 1:
return False
return True