目录
一 、题目
给你一个长度为 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
三、思路
遍历数组,该数组需满足 nums[i] <= nums[i + 1] 才符合题意中非递减数组。
若 nums[i] > nums[i + 1],那么我们需要判断 nums[i - 1] 和 nums[i + 1] 是否满足 nums[i - 1] <= nums[i + 1]
若满足上述不等式,则无需进行任何操作;反之,不满足,则令 nums[i + 1] = nums[i]。
注:nums[i] > nums[i + 1] 只能出现一次这样的情况,因为题中说明仅可以修改一次。
四、代码
class Solution:
def checkPossibility(self, nums) -> bool:
"""
:param nums: List[int]
:return: bool
"""
count = 0
for i in range(len(nums) - 1):
if nums[i] > nums[i + 1]:
count += 1
if count > 1:
return False
if i > 0 and nums[i - 1] > nums[i + 1]:
nums[i + 1] = nums[i]
return True
if __name__ == '__main__':
# nums = [4, 2, 3]
# nums = [4, 2, 1]
nums = [5, 7, 1, 8]
s = Solution()
ans = s.checkPossibility(nums)
print(ans)