【LeetCode】加一

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

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

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

示例 1:

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

示例 2:

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

---

import java.util.Scanner;

public class Main {

    public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);
        int n=sc.nextInt();
        int arr[]=new int[n];
        for (int i=0;i<arr.length;i++)
            arr[i]=sc.nextInt();
        Solution.plusOne(arr);

    }
}

class Solution {
    public static int[] plusOne(int[] digits) {

        for (int i=digits.length-1;i>=0;i--)//判断是否进位
            if(digits[i]==9)
            {
                digits[i]=0;
            }
            else
            {
                digits[i]=digits[i]+1;
                break;
            }

            if(digits[0]==0)//如果第一位为0，则建立一个比之前数组多一位的新数组
            {
                int [] newDigits=new int [digits.length+1];
                newDigits[0]=1;
                return newDigits;
            }

        return digits;
    }

}