加一 LeetCode Java

在LeetCode上，有一道非常简单的题目——【加一】。这道题目给出了一个由数字组成的列表，其中每个元素都是0-9之间的整数。我们需要将该列表看作一个数，加1后返回。

题目描述

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

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

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

示例 1：

输入：digits = [1,2,3]
输出：[1,2,4]
解释：输入数组表示数字123，我们将其加1，得到结果124。

示例 2：

输入：digits = [4,3,2,1]
输出：[4,3,2,2]
解释：输入数组表示数字4321，我们将其加1，得到结果4322。

示例 3：

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

解题思路

对于这道题目，我们只需要从列表的最后一位开始，将该位加1，并判断是否需要进位，如果需要进位，则将前一位加1，并继续判断，以此类推。

最后需要注意的是，如果最高位需要进位，则需要扩充列表的长度。

代码如下：

class Solution {
    public int[] plusOne(int[] digits) {
        int n = digits.length;
        for(int i=n-1; i>=0; i--) {
            if(digits[i] < 9) {
                digits[i]++; // 不进位
                return digits;
            } else {
                digits[i] = 0; // 进位
            }
        }
        // 如果最高位需要进位，需要扩充列表的长度
        int[] res = new int[n+1];
        res[0] = 1;
        return res;
    }
}

总结

本题虽简单，但是仍然是一道很好的练习数组操作的题目，对于Java的数组操作有一定的考验。对于懂Java基本语法的程序员而言，解决这道题目应该不是问题。