1、题目描述
https://leetcode-cn.com/problems/non-decreasing-array/
给你一个长度为 n 的整数数组,请你判断在 最多 改变 1 个元素的情况下,该数组能否变成一个非递减数列。
定义一个非递减数列的: 对于数组中任意的 i (0 <= i <= n-2),总满足 nums[i] <= nums[i + 1]。
输入: nums = [4,2,3]
输出: true
解释: 你可以通过把第一个4变成1来使得它成为一个非递减数列。
输入: nums = [4,2,1]
输出: false
解释: 你不能在只改变一个元素的情况下将其变为非递减数列。
2、代码详解
分类讨论,归纳题
- 例①:
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] 。
from typing import List
class Solution:
def checkPossibility(self, nums: List[int]) -> bool:
n = len(nums)
count = 0
for i in range(1, n):
if nums[i] < nums[i-1]:
count += 1
if i == 1 or nums[i] >= nums[i-2]:
# 1, 4, 2, 5
# 可以 把 4 调小到 1、2 或者 把 2 调大到 4、5
nums[i-1] = nums[i]
else:
# nums[i] < nums[i-2]
# 3, 4, 2, 5
# 必须 把 2 调大到 4、5
nums[i] = nums[i-1]
return count <= 1