题目:
给你一个长度为 n 的整数数组,请你判断在 最多 改变 1 个元素的情况下,该数组能否变成一个非递减数列。
我们是这样定义一个非递减数列的: 对于数组中所有的 i (1 <= i < n),总满足 array[i] <= array[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] 时,需要考虑的是应该修改数组的哪个数,使得本次修改能使 i 之前的数组成为非递减数组,并且 不影响后续的操作 。优先考虑令 nums[i - 1] = nums[i],因为如果修改 nums[i] = nums[i - 1] 的话,那么 nums[i] 这个数会变大,就有可能比 nums[i + 1] 大,从而影响了后续操作。还有一个比较特别的情况就是 nums[i] < nums[i - 2],修改 nums[i - 1] = nums[i] 不能使数组成为非递减数组,只能修改 nums[i] = nums[i - 1]。(本人代码有出入)
执行用时和内存消耗:
解题代码:
//只需提交checkPossibility()内部代码即可,main函数用来测试。
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
bool checkPossibility(vector<int>& nums) {
int cnt = 0, count = 0, f = 0,tp;
for (int i = 1; i < nums.size(); ++i) {
cnt++;
if (nums[i] < nums[i - 1]) {
tp = nums[i - 1];
nums[i - 1] = nums[i];
count++;
f = 1;
}
if (cnt > 1 && nums[i] < nums[i - 2]) {
nums[i - 2] = nums[i];
count++;
if (f == 1) {
count--;
nums[i - 1] = tp;
}
}
if (count > 1) {
return false;
}
f = 0;
}
return true;
}
int main() {
int n,t;
cin >> n;
vector<int> nums;
for (int i = 0; i < n; ++i) {
cin >> t;
nums.push_back(t);
}
if (checkPossibility(nums))
cout << "yes";
else
cout << "no";
return 0;
}