注意:数列递增递减问题,比如递增,首先想到
nums[i] >= nums[i-1]
给你一个长度为 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 <= n <= 10 ^ 4
10 ^ 5 <= nums[i] <= 10 ^ 5
思路分析:
例①: 4, 2, 5
例②: 1, 4, 2, 5
例③: 3, 4, 2, 5
当
nums[i]
破坏了数组的递增,即nums[i] < nums[i - 1]
时,为了让数组有序,我们发现一个规律(在上面三个例子中, nums[i] 都为 2, nums[i -1] 都为 4):
- 如例①的情况,当 i = 1 ,那么修改 num[i- 1] ,不要动 nums[i] ,因为nums[i]后面的元素是啥我们还不知道呢,少动它为妙。
- 如例②的情况,当 i > 1 时,我们应该优先考虑把 nums[i - 1] 调小到 >= nums[i - 2] 并且 <= nums[i]。同样尽量不去修改 nums[i] ,理由同上。
- 如例③的情况,当 i > 1 且 nums[i] < nums[i - 2] 时,我们无法调整 nums[i - 1] ,我们只能调整 nums[i] 到 nums[i - 1] 。
其实只是当发现有下降的时候,多判断了一次 nums[i]
和 nums[i - 2]
。
class Solution:
def checkPossibility(self, nums: List[int]) -> bool:
count = 0
for i in range(1, len(nums)):
if nums[i] < nums[i - 1]:
count += 1
#此处1的话nums[i-2]不会被执行,所以不会出现数组越界的情况,如果nums[i-2]被执行说明i>=2
if i == 1 or nums[i] >= nums[i - 2]:
nums[i - 1] = nums[i]
else:
nums[i] = nums[i - 1]
#最多只能有一次下降才能修改为非递减数列
return count <= 1
另一种分析:
这道题一不小心就掉坑里出不来了,我们直接看两个例子。
-
例一
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]
count = 0
for i in range(len(nums)-1):
if nums[i] > nums[i+1]:
count += 1
# 修改num[i]会导致nums[i-1] > nums[i]
if i > 0 and nums[i-1] > nums[i+1]:
nums[i+1] = nums[i]
else:
nums[i] = nums[i+1]
return count <= 1
- 总结:
- 使用
i-1
,遍历的时候从1
开始,防止越界;- 使用
i+1
,遍历的时候到len(nums) - 1
结束,防止越界。