给定一个长度为 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. 碰到前项比后项大的时候,考虑把前项变成后项,还是把后项变成前项。
2. 所以我一开始写代码是写成这样的——
class Solution {
public:
bool checkPossibility(vector& nums) {
int cnt = 0;
bool flag1;
bool flag2;
vector data1 = nums;
vector data2 = nums;
for(int i = 1;i < data1.size();i++)//前小后大
{
if(data1[i-1] > data1[i])
{
data1[i] = data1[i-1];//这个肯定没毛病
cnt++;
}
if(cnt>1)
{
flag1 = false;
break;
}
else
{
flag1 = true;
}
}
cnt = 0;
for(int i = 1;i < data2.size();i++)//前小后大
{
if(data2[i-1] > data2[i])
{
if(i>=2&&data2[i] >= data2[i-2])
data2[i-1] = data2[i];//这边可能有毛病
else
data2[i] = data2[i-1];
cnt++;
}
if(cnt>1)
{
flag2 = false;
break;
}
else
{
flag2 = true;
}
}
return flag1||flag2;
}
};
3. 对两种情况进行了分类,最后判断一下,只要有一种情况通过就说明通过。但是实际提交这段代码,发现不能通过?毛病在哪里呢?在于
if(i>=2&&data2[i] >= data2[i-2])
这个判断语句没有考虑i==1的时候的情况,因此需要改动。
4. 最后,精简后的代码如下。
class Solution {
public:
bool checkPossibility(vector& nums) {
int cnt = 0;
bool flag;
for(int i = 1;i < nums.size();i++)//前小后大
{
if(nums[i-1] > nums[i])
{
if(i>=2&&nums[i] >= nums[i-2])
nums[i-1] = nums[i];//这边可能有毛病
else if(i==1)
nums[i-1] = nums[i];
else
nums[i] = nums[i-1];
cnt++;
}
if(cnt>1)
{
flag = false;
break;
}
else
{
flag = true;
}
}
return flag;
}
};