LeetCode:66. 加一(C++)
阿新 • • 發佈:2019-02-16
題目:
給定一個由整數組成的非空陣列所表示的非負整數,在該數的基礎上加一。
最高位數字存放在陣列的首位, 陣列中每個元素只儲存一個數字。
你可以假設除了整數 0 之外,這個整數不會以零開頭。
示例 1:
輸入: [1,2,3] 輸出: [1,2,4] 解釋: 輸入陣列表示數字 123。
示例 2:
輸入: [4,3,2,1] 輸出: [4,3,2,2] 解釋: 輸入陣列表示數字 4321。
解答:
簡單題,注意判斷進位。
class Solution { public: vector<int> plusOne(vector<int>& digits) { int i = digits.size() - 1; for (; digits[i] == 9 && i > 0; i--) { digits[i] = 0; } if (i == 0 && digits[0] == 9) { digits[0] = 0; digits.insert(digits.begin(), 1); } else { digits[i]++; } return digits; } };