66_plus_one

66. 加一 简单

给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。

最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1：

输入：digits = [1,2,3]
输出：[1,2,4]
解释：输入数组表示数字 123。

示例 2：

输入：digits = [4,3,2,1]
输出：[4,3,2,2]
解释：输入数组表示数字 4321。

示例 3：

输入：digits = [0]
输出：[1]

提示：

• 1 <= digits.length <= 100
• 0 <= digits[i] <= 9

代码参考：

package main

import (
    "fmt"
)

func main() {
    // fmt.Println(plusOne([]int{7, 2, 8, 5, 0, 9, 1, 2, 9, 5, 3, 6, 6, 7, 3, 2, 8, 4, 3, 7, 9, 5, 7, 7, 4, 7, 4, 9, 4, 7, 0, 1, 1, 1, 7, 4, 0, 0, 6}))
    fmt.Println(plusOne([]int{4, 3, 2, 1}))
    fmt.Println(plusOne([]int{9}))
}

// 若用 10 的次方做位数，聚合后再拆解，不确定长度的整数一定要考虑溢出，一旦溢出结果必定出错
func plusOne(digits []int) []int {
    n := len(digits)
    if digits[n-1]+1 < 10 {
        digits[n-1]++
    } else {
        for i := n - 1; i >= 0; i-- {
            if digits[i]+1 < 10 {
                digits[i]++
                break
            }
            digits[i] = 0
            if i == 0 {
                digits = append([]int{1}, digits...)
            }
        }
    }
    return digits
}