1. 程式人生 > >【leetcode陣列和字串】加一

【leetcode陣列和字串】加一

給定一個由整陣列成的非空陣列所表示的非負整數,在該數的基礎上加一。
最高位數字存放在陣列的首位, 陣列中每個元素只儲存一個數字。
你可以假設除了整數 0 之外,這個整數不會以零開頭。

示例 1:

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

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

最高位進位
若最高位進位,那麼比他低的位數字都為9,且加1後都為0,需要初始化一個長度為(lenght+1)的新陣列,0位置為1代表進位。

最高位不進位
若最高位不進位,那麼不需要產生新陣列,後續數字由更低位計算而來。

C++做法

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        
        int carry = 1;
        for(int i = digits.size() - 1; i >= 0; i--){
            if(carry == 0)
                return digits;
            int tmp = digits[i] + carry;
            carry =
tmp / 10 ; //個位數的和 大於或等於10 進位carry都是1,否則是0 digits[i] = tmp % 10 ; } if(carry != 0){ vector<int> v(digits.size() + 1); v[0] = 1; return v; } return digits; //leetcode提交的時候不能少 } };

總結:學會了進位的演算法過程。