LeetCode 066.Plus One

题目描述

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。

最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/plus-one

思路

主要是需要考虑到进位的问题，如999进位应该为1000。

c++

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        int len = digits.size();
        int curr = len - 1;
        while(curr >= 0){
            if(digits[curr] < 9){
                digits[curr] += 1;
                return digits;
            }else{
                digits[curr] = 0;
                curr -= 1;
            }
        }
        if (curr == -1)  digits.insert(digits.begin(), 1);
        return digits;
    }
};

class Solution {
public:
    vector<int> plusOne(vector<int>& digits) {
        add(digits, 1);
        return digits;
    }
private:
    vector<int> add(vector<int>& digits, int num){
        int c = num;
        // 逆序迭代器
        for(auto it = digits.rbegin(); it != digits.rend(); ++it){
            *it += c;
            // 进位
            c = *it / 10;
            *it %= 10;
        }
        if(c > 0)   digits.insert(digits.begin(), 1);
        return digits;
    }
};