665. 非递减数列 简单

给你一个长度为 n 的整数数组,请你判断在 最多 改变 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
解释: 你不能在只改变一个元素的情况下将其变为非递减数列。

提示:

  • 1 <= n <= 10 ^ 4
    • 10 ^ 5 <= nums[i] <= 10 ^ 5

代码参考:

package main

import "fmt"

func main() {
    fmt.Println(checkPossibility([]int{3, 4, 2, 3})) // false
    fmt.Println(checkPossibility([]int{1, 4, 2, 3})) // true
}

func checkPossibility(nums []int) bool {
    smallCount := 0
    for i := 1; i < len(nums); i++ {
        if nums[i-1] > nums[i] {
            if smallCount == 1 {
                return false
            }

            // 巧妙一步:如果 i-2 的数比 i 的大,就向后延续大数的影响范围
            // 3 4 2 3    // 3 4 4 3
            // 1 4 2 3    // 1 4 2 3
            if i >= 2 && nums[i-2] >= nums[i] {
                nums[i] = nums[i-1]
            }
            // fmt.Println(nums)
            smallCount++
        }
    }
    return true
}
最后编辑: kuteng  文档更新时间: 2021-06-05 10:16   作者:kuteng