luoheng
10/7/2019 - 1:52 AM

plusOne

func plusOne(digits []int) []int {
    plus, n := 1, len(digits)-1
    for plus == 1 && n >= 0 {
        if digits[n] == 9 {
            digits[n] = 0
            n--
        } else {
            digits[n] += 1
            plus = 0
        }
    }
    if n == -1 {
        newD := make([]int, len(digits)+1)
        newD[0] = 1
        copy(newD[1:], digits)
        return newD
    }
    return digits
}