[LeetCode] Plus One 加一運算

Given a non-negative number represented as an array of digits, plus one to the number.

The digits are stored such that the most significant digit is at the head of the list.

 

將一個數字的每個位上的數字分別存到一個一維向量中，最高位在最開頭，我們需要給這個數字加一，即在末尾數字加一，如果末尾數字是9，那么則會有進位問題，而如果前面位上的數字仍為9，則需要繼續向前進位。具體算法如下：首先判斷最后一位是否為9，若不是，直接加一返回，若是，則該位賦0，再繼續查前一位，同樣的方法，知道查完第一位。如果第一位原本為9，加一后會產生新的一位，那么最后要做的是，查運算完的第一位是否為0，如果是，則在最前頭加一個1。代碼如下：

 

C++ 解法一：

class Solution {
public:
    vector<int> plusOne(vector<int> &digits) {
        int n = digits.size();
        for (int i = n - 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;
    }
};

 

Java 解法一：

public class Solution {
    public int[] plusOne(int[] digits) {
        int n = digits.length;
        for (int i = digits.length - 1; i >= 0; --i) {
            if (digits[i] < 9) {
                ++digits[i];
                return digits;
            }
            digits[i] = 0;
        }
        int[] res = new int[n + 1];
        res[0] = 1;
        return res;
    }
}

 

我們也可以使用跟之前那道Add Binary類似的做法，我們將carry初始化為1，然后相當於digits加了一個0，處理方法跟之前那道題一樣，參見代碼如下：

 

C++ 解法二 ：

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        if (digits.empty()) return digits;
        int carry = 1, n = digits.size();
        for (int i = n - 1; i >= 0; --i) {
            if (carry == 0) return digits;
            int sum = digits[i] + carry;
            digits[i] = sum % 10;
            carry = sum / 10;
        }
        if (carry == 1) digits.insert(digits.begin(), 1);
        return digits;
    }
};

 

Java 解法二 ：

public class Solution {
    public int[] plusOne(int[] digits) {
        if (digits.length == 0) return digits;
        int carry = 1, n = digits.length;
        for (int i = digits.length - 1; i >= 0; --i) {
            if (carry == 0) return digits;
            int sum = digits[i] + carry;
            digits[i] = sum % 10;
            carry = sum / 10;
        }
        int[] res = new int[n + 1];
        res[0] = 1;
        return carry == 0 ? digits : res;
    }
}

 

類似題目：

Add Binary

 

參考資料：

https://discuss.leetcode.com/topic/24288/my-simple-java-solution

 

LeetCode All in One 題目講解匯總(持續更新中...)