LeetCode:66. 加一(简单)

题目描述

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。
最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1： 输入：digits = [1,2,3] 输出：[1,2,4] 解释：输入数组表示数字 123。
示例 2： 输入：digits= [4,3,2,1] 输出：[4,3,2,2] 解释：输入数组表示数字 4321。
示例 3： 输入：digits = [0] 输出：[1]

题解

对数组中数字进行分析,加1的数字存在两种情况

1. 除9之外的数字(直接返回加1的值)
2. 数字9(需要考虑是否要进位)

class Solution {//cpp
public:
    vector<int> plusOne(vector<int>& digits) {
        for(int i = digits.size() - 1; i >= 0; i--){
            if(digits[i] != 9){
                digits[i]++;
                return digits;
            }
            digits[i] = 0;
        }
        // 若跳出for循环，说明数组元素全为9
        digits.insert(digits.begin(), 1);//数组digits中迭代器指向元素(数组头)前增加一个元素1
        return digits;
        /*
        vector res(digits.size() + 1);
        res[0] = 1;
        for(int i = 1; i < res.size(); i++){
            res[i] = 0;
        }
        return res;
        */
    }
};

class Solution:#python
    def plusOne(self, digits: List[int]) -> List[int]:
        carry = 1
        for i in range(len(digits) - 1, -1, -1):
            digits[i], carry = (carry + digits[i]) % 10, (carry + digits[i]) // 10 #求余, 向下整除
            #将第i位数字加1再更新,如果第i位加1为10,则需要进位,将carry置为1,这样再往前遍历时前1位数字加1,如果不需要进位,carry置为0,往前遍历时前1位数字不变
        return [carry] + digits if carry else digits
        #如果carry为1,说明最高位为9(原数组所有元素为9)且加1变为0,需要在最前面增加一位1

参考

数组加一

相关题目

面试题 02.05. 链表求和