【Leetcode】66. 加一

作者: 码蹄疾
毕业于哈尔滨工业大学。小米广告第三代广告引擎的设计者、开发者；
负责小米应用商店、日历、开屏广告业务线研发；
主导小米广告引擎多个模块重构；
关注推荐、搜索、广告领域相关知识;

题目

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

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

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

示例 1:

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

示例 2:

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

题解

这个题目直接模拟就好，注意一下进位的情况：

模拟[1,2,3,9]

[1,2,3,9] 进位:0
[1,2,3,0] 进位:1
[1,2,4,0] 进位:0

模拟[9,9]

[9,9] 进位:0
[9,0] 进位:1
[0,0] 进位:1
[1,0,0]

python

class Solution:
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        carry = 1
        res = []
        for digit in reversed(digits):
            sum = digit + carry
            res.append(int(sum % 10))
            carry = int(sum / 10)
        if carry != 0:
            res.append(carry)
        res.reverse()
        return res

java

class Solution {
    public int[] plusOne(int[] digits) {
        int carry = 1;
        for (int i = digits.length - 1; i >= 0; i--) {
            int sum = digits[i] + carry;
            digits[i] = sum % 10;
            carry = sum / 10;
        }
        if (carry != 0) {
            int[] b = new int[digits.length + 1];
            b[0] = carry;
            System.arraycopy(digits, 0, b, 1, digits.length);
            return b;
        }
        return digits;
    }
}