Leetcode解题之路(golang版)：66. 加一(Plus One)

66. 加一

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

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

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

示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。

示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

解法一：

空间复杂度：$O(n)$

时间复杂度：$O(n)$

//仔细想一下，这道题其实分为三种情况
//1. 首先是数组最后一位不为9，也就是不产生进位的情况，直接把最后一位加一就可以
//2. 后边为9但是最高位不是9，这样的会产生进位，但是不需要最高位不进位，不需要扩容slice
//3. 最后就是slice都是9的情况，需要扩容
func plusOne(digits []int) []int {
    //情况1
    if digits[len(digits) - 1] != 9 {
        digits[len(digits) - 1]++
        return digits
    }
    loc := len(digits) - 1
    //情况2和3 要把9进位变为0
    for ; loc >= 0 && digits[loc] == 9; loc-- {
        digits[loc] = 0
    }
    //如果loc没有到-1 说明最高位不是9 不需要扩容
    if loc >= 0 {
        digits[loc]++
    //需要扩容的情况
    } else {
        digits = append([]int{1}, digits...)
    }
    return digits
}