题目描述

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

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

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

示例:

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

思路

1.我们可以从数组的最后一位开始加一,加一后将值与10进行mod运算,若取模后等于0,说明要进行进位操作,继续进行循环即可;若不等于0,说明不需要进位,直接返回即可。
2.若我们从循环出来后还未返回结果,说明还需要进位操作,我们初始化一个长度比之前数组大一的新数组,然后将数组的第一个位置赋值为1即可,因为数组的默认值是0,是与原数组相同的。

Java代码实现

    public int[] plusOne(int[] digits) {
        int len = digits.length;
        for(int i=len-1;i>=0;i--){
            digits[i] = digits[i] + 1;
            digits[i] = digits[i]%10;
            if(digits[i] != 0)
                return digits;
        }
        digits = new int[len + 1];
        digits[0] = 1;
        return digits;
    }

Golang代码实现

func plusOne(digits []int) []int {
    if len(digits) == 0{
        return digits
    }
    length := len(digits)
    for i:= length-1;i>=0;i--{
        cur := digits[i]+1
        digits[i] = cur%10
        if digits[i] != 0{
            return digits
        }
    }

    digits = make([]int,length+1)
    digits[0] = 1

    return digits;
}