给你一个长度为 n 的整数数组,请你判断在 最多 改变 1 个元素的情况下,该数组能否变成一个非递减数列。
我们是这样定义一个非递减数列的: 对于数组中所有的 i (0 <= i <= n-2),总满足 nums[i] <= nums[i + 1]。
示例 1:
输入: nums = [4,2,3]
输出: true
解释: 你可以通过把第一个4变成1来使得它成为一个非递减数列。
示例 2:
输入: nums = [4,2,1]
输出: false
解释: 你不能在只改变一个元素的情况下将其变为非递减数列。
题解:这道题一不小心就掉坑里出不来了,我们直接看两个例子。
例一
1,4,2,3
1,4,2,3
我们发现 4 比 2 大,那么把 4 替换成 2,变成 1,2,2,3 就好了
但是如果把 2 替换成 4,变成 1,4,4,3 是不可以的
例二
5,7,1,8
5,7,1,8
我们发现 7 比 1 大,那么把 1 替换成 7 变成 5, 7, 7, 8 就好了
但是如果把 7 替换成 1,变成 5,1,1,8 是不可以的
于是,对于 nums[i] > nums[i+1], 在第一个例子里我们修改的是前一个数,第二个例子里我们修改的是后一个数。
看出什么了没有?
如果发现 nums[i] > nums[i+1],我们一定是通过修改其中一个元素,让 nums[0] 到 nums[i+1] 满足非递减数列,且不干扰到后面的大小关系。而不是拆了东墙补西墙,就像 1,4,2,3 变成 1,4,4,3 这样。
所以我们的策略是:
能修改 nums[i],就绝不修改 nums[i+1]
class Solution:
def checkPossibility(self, nums: List[int]) -> bool:
changed = False
for i in range(len(nums)-1):
if nums[i] > nums[i+1]:
if changed:
return False
changed = True
# 修改num[i]会导致nums[i-1] > nums[i]
if i > 0 and nums[i-1] > nums[i+1]:
nums[i+1] = nums[i]
else:
#能修改i绝不修改i+1
nums[i] = nums[i+1]
return True