1. 程式人生 > >【演算法題】加一

【演算法題】加一

/*
給定一個由整陣列成的非空陣列所表示的非負整數,在該數的基礎上加一。

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

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

示例 1:

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

輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入陣列表示數字 4321。
 */
public class Test07
{
    public static int[] plusOne(int[] digits)
    {
        for (int i = digits.length - 1; i >= 0; i--) {
            if (digits[i] == 9) {
                digits[i] = 0;
                continue;
            } else {
                digits[i] = digits[i] + 1;
                break;
            }
        }
        if (digits[0] == 0) {
            int[] aaa = new int[digits.length+1];
            aaa[0] = 1;
            return aaa;
        }
        return digits;
    }

    public static void main(String[] args)
    {
        int[] aaa = {9, 9, 9, 9};
        int[] ints = plusOne(aaa);
        for (int anInt : ints) {
            System.out.print(anInt + ",");
        }
    }
}

這道題主要難點在於數字為9的時候帶來的進位問題,意識到這一點這道題就很好解決了.我採用的方法是迴圈遍歷判斷當前位數是否為9,是則變為0,繼續判斷前一位,直到不為9則加1.若全為9,則遍歷完成陣列元素就會全為0,此時new一個長度加1的陣列,首位設為1即可