3、加一

题目描述

66. 加一

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

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

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

示例 1：

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

示例 2：

输入：digits = [4,3,2,1] 输出：[4,3,2,2] 解释：输入数组表示数字 4321。

示例 3：

输入：digits = [0] 输出：[1]

数组遍历

题目解析

我们思考一下，加一的情况一共有几种情况呢？是不是有以下三种情况

加一

则我们根据什么来判断属于第几种情况呢？

我们可以根据当前位 余 10 来判断，这样我们就可以区分属于第几种情况了，大家直接看代码吧，很容易理解的。

代码

Java Code:

class Solution {
    public int[] plusOne(int[] digits) {
        //获取长度
        int len = digits.length;
        for (int i = len-1; i >= 0; i--) {
            digits[i] = (digits[i] + 1) % 10;
            //第一种和第二种情况，如果此时某一位不为 0 ，则直接返回即可。
            if (digits[i] != 0) {
                return digits;
            }

        }
        //第三种情况，因为数组初始化每一位都为0，我们只需将首位设为1即可
        int[] arr = new int[len+1];
        arr[0] = 1;
        return arr;
    }
}

Python Code:

from typing import List
class Solution:
    def plusOne(self, digits: List[int])->List[int]:
        # 获取长度
        leng = len(digits)
        for i in range(leng - 1, -1, -1):
            digits[i] = (digits[i] + 1) % 10
            # 第一种和第二种情况，如果此时某一位不为 0 ，则直接返回即可。
            if digits[i] != 0:
                return digits
        # 第三种情况，因为数组初始化每一位都为0，我们只需将首位设为1即可
        arr = [0] * (leng + 1)
        arr[0] = 1
        return arr

C++ Code:

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        for(int i = digits.size() - 1; i >= 0; --i){
            digits[i] = (digits[i] + 1)%10;
            if(digits[i]) return digits;
        }
        for(int & x: digits) x = 0;
        digits.emplace_back(1);
        reverse(digits.begin(), digits.end());
        return digits;
    }
};

Swift Code:

class Solution {
    func plusOne(_ digits: [Int]) -> [Int] {
        let count = digits.count
        var digits = digits
        for i in stride(from: count - 1, through: 0, by: -1) {
            digits[i] = (digits[i] + 1) % 10
            if digits[i] != 0 {
                return digits
            }
        }
        var arr: [Int] = Array.init(repeating: 0, count: count + 1)
        arr[0] = 1
        return arr
    }
}

Go Code:

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