1. 程式人生 > >【leetcode之旅】陣列- 66. 加一問題

【leetcode之旅】陣列- 66. 加一問題

給定一個非負整數組成的非空陣列,在該數的基礎上加一,返回一個新的陣列。

最高位數字存放在陣列的首位, 陣列中每個元素只儲存一個數字。

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

示例 1:

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

示例 2:

輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入陣列表示數字 4321。
public class PlusOne {
    public static void main(String[] args) {
        int[] nums = {9,8,9,9};
        int[] results = plusOne1(nums);
        for (int i = 0; i < results.length; i++) {
            System.out.print(results[i]);
        }
    }

    /**
     * 方法一:
     * @param digits
     * @return
     */
    public static int[] plusOne1(int[] digits){
        int len = digits.length;
        for (int i = len - 1; i >= 0; i--) {
            if (digits[i] == 9){
                digits[i] = 0;
            }else{
                digits[i]++;
                return digits;
            }
        }
        int[] results = new int[len + 1];
        results[0] = 1;
        return results;
    }

    /**
     * 方法二:
     * @param digits
     * @return
     */
    public static int[] plusOne2(int[] digits){
        int plus = 1;
        for (int i = digits.length - 1; i >= 0; i--) {
            if (plus == 0) {
                return digits;
            }
            int tmp = digits[i] + plus;
            //若進位則為1,,否則為0
            plus = tmp / 10;
            //若進位則為0,,否則加一
            digits[i] = tmp % 10;
        }
        //如果每位數都為9,則需要擴容
        if (plus != 0) {
            int[] result = new int[digits.length + 1];
            result[0] = 1;
            return result;
        }
        return digits;
    }
}