1. 程式人生 > >陣列和字串//加一

陣列和字串//加一

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

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

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

示例 1:

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

示例 2:

輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入陣列表示數字 4321。
class Solution {
    public int[] plusOne(int[] digits) {
        int carry = 1;
        for(int i = digits.length - 1; i >= 0; i--){
            if(carry == 0){
                return digits;
            }
            int temp = digits[i] + carry;
            carry = temp/10;
            digits[i] = temp%10;
        }
        if(carry != 0){
            int[] result = new int[digits.length+1];
            result[0] = 1;
            return result;
        }
        return digits;
    }
}
class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        if(digits.empty()) return digits;
        int carry = 1;
        int 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;
    }
};
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;
    }
};