LeetCode 66.加一

加一

給定一個由整數組成的非空數組所表示的非負整數，在該數的基礎上加一。

最高位數字存放在數組的首位， 數組中每個元素只存儲一個數字。

你可以假設除了整數 0 之外，這個整數不會以零開頭。

示例 1:

輸入: [1,2,3]
輸出: [1,2,4]
解釋: 輸入數組表示數字 123。

示例 2:

輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入數組表示數字 4321。

解題思路

數組最後一位是個位，處理時從後向前進行，要注意最後爲9時需要進位，第一位爲9時需要增加數組的大小。

AC代碼

public int[] plusOne(int[] digits) {
        int sum = 0;
        int carry = 0;
        for (int i = digits.length - 1; i >= 0; i--) {
            sum = digits[i] + 1;
            carry = sum / 10;
            digits[i] = sum % 10;
            if (carry == 0) break;
        }
        if (carry == 0) {
            return digits;
        } else {
            int[] result = new int[digits.length + 1];
            result[0] = carry;
            for (int i = 1; i < result.length; i++) {
                result[i] = digits[i - 1];
            }
            return result;
        }
    }