题目原文
给定一个长度为 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.数组个数小于两个,返回True;
2.思路如下:
如果出现 a[i] > a[i+1] ,改变一个数,就面临两种选择:
1)把a[i]变大;
2)把a[i+1] 变小。
这两种选择其实是有依据的,需要比较a[i-1] 与 a[i+1]的值:
eg. … 1 4 3 … 只能选择把4变小 … 3 4 1 … 只能选择把1变大
改变完之后,记录改变次数,再检测是否升序,如果次数大于1,至少改了两次,返回false。
class Solution:
def checkPossibility(self, nums: List[int]) -> bool:
if len(nums) < 3:
return True
count = 0
if nums[0] > nums[1]:
count += 1
nums[0] = nums[1]
for i in range(1,len(nums)-1):
if nums[i] > nums[i+1]:
count += 1
if count > 1:
return False
if nums[i-1] > nums[i+1]:
nums[i+1] = nums[i]
else:
nums[i] = nums[i-1]
return count <= 1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/non-decreasing-array