题目地址:
https://leetcode.com/problems/non-decreasing-array/
给定一个长 n n n数组 A A A,问是否能只改变一个数使其成为单调增数组。这里的单调增不一定要严格。
遍历 A A A,如果发现 A [ i ] > A [ i + 1 ] A[i]>A[i+1] A[i]>A[i+1]了,则说明需要动用一次修改的权利。如果之前已经使用过一次这个权利了,那直接返回false。否则,如果 i + 1 = n − 1 i+1=n-1 i+1=n−1,那可以改 A [ i + 1 ] A[i+1] A[i+1]达到目的,所以返回true;接着讨论 A [ i + 1 ] A[i+1] A[i+1]和 A [ i + 2 ] A[i+2] A[i+2]的关系。如果 A [ i + 1 ] > A [ i + 2 ] A[i+1]>A[i+2] A[i+1]>A[i+2],则无解,返回false;如果 A [ i + 1 ] = A [ i + 2 ] A[i+1]=A[i+2] A[i+1]=A[i+2],则必须改 A [ i ] A[i] A[i],可以将其改为 A [ i + 1 ] A[i+1] A[i+1];如果 A [ i + 1 ] < A [ i + 2 ] A[i+1]<A[i+2] A[i+1]<A[i+2],此时需要看一下 A [ i ] A[i] A[i]和 A [ i + 2 ] A[i+2] A[i+2]的关系,如果 A [ i ] ≤ A [ i + 2 ] A[i]\le A[i+2] A[i]≤A[i+2],则必须改 A [ i + 1 ] A[i+1] A[i+1],可以将其改为 A [ i ] A[i] A[i];如果 A [ i ] > A [ i + 2 ] A[i]> A[i+2] A[i]>A[i+2],则必须改 A [ i ] A[i] A[i],可以将其改为 A [ i + 1 ] A[i+1] A[i+1]。遍历完之后,还需要再判断一下改动后的数组是否符合条件。代码如下:
public class Solution {
public boolean checkPossibility(int[] nums) {
if (nums.length <= 1) {
return true;
}
boolean changed = false;
for (int i = 0; i < nums.length - 1; i++) {
if (nums[i] > nums[i + 1]) {
if (changed) {
return false;
}
if (i + 1 == nums.length - 1) {
return true;
}
if (nums[i + 1] > nums[i + 2]) {
return false;
} else if (nums[i + 1] < nums[i + 2]) {
if (nums[i] <= nums[i + 2]) {
nums[i + 1] = nums[i];
} else {
nums[i] = nums[i + 1];
}
} else {
nums[i] = nums[i + 1];
}
changed = true;
}
}
// 改完再判断一下改完后的数组是否满足条件
for (int i = 0; i < nums.length - 1; i++) {
if (nums[i] > nums[i + 1]) {
return false;
}
}
return true;
}
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。