LeetCode150道面试经典题-- 加一(简单)

Alphamilk2023-08-18 14:30

1.题目

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

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

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

2.示例

示例 1：

输入：digits = [1,2,3]

输出：[1,2,4]

解释：输入数组表示数字 123。

示例 2：

输入：digits = [4,3,2,1]

输出：[4,3,2,2]

解释：输入数组表示数字 4321。

示例 3：

输入：digits = [0]

输出：[1]

3.思路

这道题看着比较简单，但是也是存在有陷阱，

第一点，当末位数是9时候，在加一时候需要进位。

其次就是当首位为9时候，加一需要重新创建一个新的数组，并且在新的数组的首位赋值为1即可

4.代码

java 复制代码

class Solution {
    public int[] plusOne(int[] digits) {
         if (digits.length==1 &&digits[0]==0){
            digits[0] = 1;
        }else {
            digits[digits.length-1]+=1;
        }
         for (int i= digits.length-1;i>=0;i--){
            if (digits[i]==10){
                if (digits.length ==1 || i == 0){
                    digits = new int[digits.length+1];
                    digits[0]=1;
                    break;
                }
                digits[i] =0;
                digits[i-1]+=1;
            }
        }
        return digits;
    }
}