给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
输入: [1,2,3] 输出: [1,2,4] 解释: 输入数组表示数字 123。 输入: [4,3,2,1] 输出: [4,3,2,2] 解释: 输入数组表示数字 4321。
理解题意:
- 其实我们可以认为就是nums[len(nums)-1] +=1
- 但是有个陷阱,那就是当参数为{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 }