Given an array with n integers, your task is to check if it could become non-decreasing by modifying at most 1 element.
We define an array is non-decreasing if array[i] <= array[i + 1] holds for every i (1 <= i < n).
Example 1:
Input: [4,2,3]
Output: True
Explanation: You could modify the first
4
to
1
to get a non-decreasing array.
Example 2:
Input: [4,2,1]
Output: False
Explanation: You can’t get a non-decreasing array by modify at most one element.
Note: The n belongs to [1, 10,000].
本来以为是个很简单的题为什么AC率那么低。。
先是尝试提交了两个答案,都fail了,。还是case写的不到位啊。
仔细考虑了一下,换了一种思路:
题目的意思就是给定一个数组,考虑能否在只修改一个值的情况下使得该数组成为一个非递减数组。
那么遍历一遍,当遇到nums[i]
bool checkPossibility(int* nums, int numsSize) {
int A=0,B=0,index=1;/*A,B分别表示修改nums[i]和nums[i-1]成功与否,index表示修改的次数*/
int tmp;
if(numsSize<3)return true;
for(int i=1;i<numsSize;i++)
{
/*修改nums[i]或者nums[i-1]均不可保持非递减*/
if(((A+B)==2)||(index<0))return false;
if(nums[i]<nums[i-1])
{
tmp = nums[i];
nums[i] = nums[i-1];
if(i<numsSize-1 && nums[i]>nums[i+1])
A=1;
else//修改nums[i]成功,则index--,A和B恢复0进行下一次比较
{
index--;
A=0;
B=0;
continue;
}
nums[i] = tmp;
nums[i-1] = nums[i];
if(i>1&&(nums[i-1]<nums[i-2]))
B=1;
else
{
index--;
A=0;
B=0;
continue;
}
}
}
/*边界情况如case 1,5,4,6,2*/
if(((A+B)==2)||(index<0))return false;
else
return true;
}
注意循环的边界,以及case的设计。
优化后的代码:
bool checkPossibility(int* nums, int numsSize)
{
int index = 1;
for(int i=0;i<numsSize && index>0;i++)
{
if(nums[i]<nums[i-1])
{
index--;
if(i>2 && nums[i]>nums[i-2])nums[i-1]=nums[i];
else
nums[i] = nums[i-1];
}
}
}