leetcode 66. 加一(Plus One)

Posted 2021-02-13 zhanzq1

目录

• 题目描述：
• 示例 1:
• 示例 2:
• 解法：

题目描述：

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

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

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

示例 1:

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

示例 2:

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

---

解法：

class Solution {
public:
    vector plusOne(vector& digits) {
        vector res = digits;
        res.insert(res.begin(), 0);
        int sz = res.size();
        int carry = 1;
        for(int i = sz-1; i >= 0; i--){
            res[i] += carry;
            if(res[i] >= 10){
                carry = 1;
                res[i] -= 10;
            }else{
                carry = 0;
            }
        }
        if(res[0] == 0){
            res.erase(res.begin());
        }
        return res;
    }
};