加一的golang实现

Posted 2021-01-22 timliudream

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

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

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

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

理解题意：

1. 其实我们可以认为就是nums[len(nums)-1] +=1
2. 但是有个陷阱，那就是当参数为{9}的时候，输出是{1,0}，那这样就有点麻烦了。

解题思路：

那我们就可以分为两种情况：

• 第一种就是当前位加1不等于10
• 第二种就是当前位加1等于10

但是也要注意当下标为0的值为9的时候要加1，那我们切片的长度是不够的，我们只能用append来扩容

具体代码：

func plusOne(digits []int) []int {
    //标志位，当前下标的上一位是否要加一
    isPlusOne := true
    for i := len(digits) - 1; i >= 0; i-- {
        if isPlusOne {
            digits[i] += 1
            if digits[i] == 10 {
                //这里还需要判断是不是已经到0这个下标了，当是0这个下标，但是还需要加1的话，
                //那就要在切片的最前面插入
                digits[i] = 0
                isPlusOne = true
                if i==0{//已经到了0这个小标
                    s:=[]int{1}
                    digits = append(s,digits...)
                }
                continue
            }
            isPlusOne = false
        }
    }
    return digits
}