LeetCode Array Easy 66. Plus One

Description

Given a non-empty array of digits representing a non-negative integer, plus one to the integer.

The digits are stored such that the most significant digit is at the head of the list, and each element in the array contain a single digit.

You may assume the integer does not contain any leading zero, except the number 0 itself.

Example 1:

Input: [1,2,3]
Output: [1,2,4]
Explanation: The array represents the integer 123.

Example 2:

Input: [4,3,2,1]
Output: [4,3,2,2]
Explanation: The array represents the integer 4321.

　　题目理解：输入一个数组，数组中的每个元素都不大于10，每一个元素相当于一个正整数每一位的数，输出这个正整数加一之后的数，用数组表示

　　分析：1，如果当前为不为9则直接ji当前位加1即可

　　　　　2，如果为9 则当前位置为0，继续1的操作。

我的解决方法

public class Solution {
    public int[] PlusOne(int[] digits) {
            int[] result = null;
            int i = digits.Length - 1;
            while (i >= 0 && digits[i] +1 == 10)
            {
                digits[i] = 0;
                i--;
            }
            if (i >= 0)
            {
                digits[i] = digits[i] + 1;
                return digits;
            }
            result = new int[digits.Length +1];
            
            result[0] = 1;
            return result;
        
    }
}

288ms 

另一种更好的解决方案（264ms）

public 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;
            }
            digits[i]= 0;
        }
        
        int[] answer = new int[n+1];
        answer[0] = 1;
        return answer;
    }
}