1. 程式人生 > >leetcode(66)加一

leetcode(66)加一

題目描述
給定一個由整陣列成的非空陣列所表示的非負整數,在該數的基礎上加一。
最高位數字存放在陣列的首位, 陣列中每個元素只儲存一個數字。
你可以假設除了整數 0 之外,這個整數不會以零開頭。
示例 1:
輸入: [1,2,3]
輸出: [1,2,4]
解釋: 輸入陣列表示數字 123。
示例 2:
輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入陣列表示數字 4321。

解題思路
這個題例子其實給少了,就是產生進位的情況,例如輸入[9]的時候,加一的話意味著產生一個僅為,輸出矩陣變為[1,0],所以這個可以分為兩種情況,一就是無進位的時候,二就是有進位的時候。
無進位的時候,只需最後一位加一即可,當有進位的時候,需要產生新的陣列,承載進位之後多出來的數值。
程式碼如下

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int length = digits.size();
        for (int i = length - 1; i >= 0; --i) 
        {
            if (digits[i] == 9) 
                digits[i] = 0;
            else 
            {
                digits[i] += 1;
                return digits;
            }
        }
        if (digits.front() == 0) 
            digits.insert(digits.begin(), 1);
        return digits;
        
    }
};