题目
给你一个长度为 n 的整数数组 nums ,请你判断在 最多 改变 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
解释: 你不能在只改变一个元素的情况下将其变为非递减数列。
提示:
n == nums.length
1 <= n <= 104
-105 <= nums[i] <= 105
答案
class Solution {
public:
bool checkPossibility(vector<int>& nums) {
int count = 0;
for (int i = 1; i < nums.size(); i++) {
if (nums[i] < nums[i - 1]) {
if (i == 1 || nums[i] >= nums[i - 2]) {
nums[i - 1] = nums[i];
} else {
nums[i] = nums[i - 1];
}
count ++;
}
}
return count <= 1;
}
};
算法原理
- 初始化计数器:首先,定义了一个名为 count 的计数器变量,并将其初始化为 0。这个计数器用于记录修改操作的次数,以确保整个过程中只进行了至多一次的修改。
- 遍历数组:然后,使用一个 for 循环从数组的第二个元素(索引为 1)开始遍历,直到数组的末尾。这是因为我们需要比较当前元素与其前一个元素的关系。
- 检查非递减条件:在每次迭代中,检查当前元素 nums[i] 是否小于其前一个元素 nums[i - 1]。如果是,说明需要调整以满足非递减条件。
- 调整元素:
- 如果 i 等于 1(即当前元素是第二个元素),或者当前元素 nums[i] 大于或等于其前两个元素 nums[i - 2],那么说明将前一个元素 nums[i - 1] 修改为当前元素 nums[i] 是安全的,因为这样不会破坏前两个元素之间的非递减关系。执行此修改后,继续遍历。
- 否则,说明将当前元素 nums[i] 修改为前一个元素 nums[i - 1] 是更好的选择,因为这样可以确保从当前位置开始的后续元素都保持非递减关系。执行此修改后,继续遍历。
- 增加计数器:每次执行修改操作时,都将 count 计数器的值增加 1。
- 检查修改次数:遍历完成后,检查 count 的值是否小于或等于 1。如果是,说明在整个过程中只进行了至多一次的修改,因此数组可以通过至多一次修改成为非递减序列,函数返回 true。否则,如果 count 的值大于 1,说明进行了超过一次的修改,函数返回 false。