加一的golang实现

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

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

你可以假设除了整数 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
}
原文地址:https://www.cnblogs.com/TimLiuDream/p/10023642.html