【Leetcode】66. 加一
作者: 碼蹄疾
畢業於哈爾濱工業大學。 小米廣告第三代廣告引擎的設計者、開發者;
負責小米應用商店、日曆、開屏廣告業務線研發;
主導小米廣告引擎多個模組重構;
關注推薦、搜尋、廣告領域相關知識;
題目
給定一個由整陣列成的非空陣列所表示的非負整數,在該數的基礎上加一。
最高位數字存放在陣列的首位, 陣列中每個元素只儲存一個數字。
你可以假設除了整數 0 之外,這個整數不會以零開頭。
示例 1:
輸入: [1,2,3] 輸出: [1,2,4] 解釋: 輸入陣列表示數字 123。
示例 2:
輸入: [4,3,2,1] 輸出: [4,3,2,2] 解釋: 輸入陣列表示數字 4321。
題解
這個題目直接模擬就好,注意一下進位的情況:
- 模擬[1,2,3,9]
[1,2,3,9] 進位:0 [1,2,3,0] 進位:1 [1,2,4,0] 進位:0
- 模擬[9,9]
[9,9] 進位:0 [9,0] 進位:1 [0,0] 進位:1 [1,0,0]
python
class Solution: def plusOne(self, digits): """ :type digits: List[int] :rtype: List[int] """ carry = 1 res = [] for digit in reversed(digits): sum = digit + carry res.append(int(sum % 10)) carry = int(sum / 10) if carry != 0: res.append(carry) res.reverse() return res
java
class Solution { public int[] plusOne(int[] digits) { int carry = 1; for (int i = digits.length - 1; i >= 0; i--) { int sum = digits[i] + carry; digits[i] = sum % 10; carry = sum / 10; } if (carry != 0) { int[] b = new int[digits.length + 1]; b[0] = carry; System.arraycopy(digits, 0, b, 1, digits.length); return b; } return digits; } }