1. 程式人生 > >Leetcode:66.加一

Leetcode:66.加一

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

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

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

示例 1:

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

示例 2:

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

解題思路:

注意幾個特殊的情況:

1. 可能存在進位

2. 可能存在長度增加

C++程式碼
class Solution { public:     vector<int> plusOne(vector<int>& digits) {         int sgn = 1, size = digits.size(), data;         for (int i = size; i >= 1; i--) {             data = digits[i - 1] + sgn;             sgn = data / 10;             digits[i - 1] = data % 10;         }         if (sgn > 0) digits.insert(digits.begin(), 1);         return digits;     } };