题目描述
给定一个长度为 n 的整数数组,判断在最多改变 1 个元素的情况下,该数组能否变成一个非递减数列。
非递减数列的: 对于数组中所有的 i (1 <= i < n),满足 array[i] <= array[i + 1]。
示例
输入: [4,2,3]
输出: True
解释: 你可以通过把第一个4变成1来使得它成为一个非递减数列。
输入: [4,2,1]
输出: False
解释: 你不能在只改变一个元素的情况下将其变为非递减数列。
解答
用count来记录调整元素的次数,若大于1则返回false,否则返回true
遍历数组,发现不满足array[i] <= array[i + 1]则调整数组元素
重点在于如何调整元素
[a,b,c]
非递减的话应当a<=b<=c
用指针i指向三个数中的中间的数,即b,且i之前的元素都满足顺序,即a<=b
如果c>b,则需要调整元素:
- c<=a,则目前顺序为c<=a<=b,故需要调整c,调整c=b
- c>a,则目前顺序为a<=b,a<c,故需调整b,调整b=c
class Solution(object):
def checkPossibility(self, nums):
"""
:type nums: List[int]
:rtype: bool
"""
length = len(nums)
if length<=2:
return True
count = 0 # 用于记录调整顺序的次数
# 保证第一第二个元素满足顺序
if nums[0]>nums[1]:
nums[0] = nums[1]
count += 1
for i in range(1,length-1):
# i之前的都满足顺序
# 第i个和第i+1个元素不满足顺序
if nums[i]>nums[i+1]:
count += 1
if count > 1:
return False
# 调整第i或第i+1个元素,使它们满足顺序
# 第i个大于第i+1个,第i+1个小于第i-1个,则第i+1个小,不满足顺序
if nums[i+1]<=nums[i-1]:
nums[i+1] = nums[i]
# 第i个大于第i+1个,第i+1个大于第i-1个,则第i个不满足顺序
else:
nums[i] = [i+1]
return True