LeetCode - 989. 数组形式的整数加法

描述

对于非负整数 X 而言,X 的数组形式是每位数字按从左到右的顺序形成的数组。例如,如果 X = 1231,那么其数组形式为 [1,2,3,1]。

给定非负整数 X 的数组形式 A,返回整数 X+K 的数组形式。

 

示例 1:

输入:A = [1,2,0,0], K = 34
输出:[1,2,3,4]
解释:1200 + 34 = 1234
示例 2:

输入:A = [2,7,4], K = 181
输出:[4,5,5]
解释:274 + 181 = 455
示例 3:

输入:A = [2,1,5], K = 806
输出:[1,0,2,1]
解释:215 + 806 = 1021
示例 4:

输入:A = [9,9,9,9,9,9,9,9,9,9], K = 1
输出:[1,0,0,0,0,0,0,0,0,0,0]
解释:9999999999 + 1 = 10000000000
 

提示:

1 <= A.length <= 10000
0 <= A[i] <= 9
0 <= K <= 10000
如果 A.length > 1,那么 A[0] != 0

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/add-to-array-form-of-integer/
 

求解

    class Solution {
    public:
        vector<int> addToArrayForm(vector<int> &A, int K) {
            const int n = A.size();
            vector<int> res;
            res.reserve(n + 1);
            int carry = 0;  // 进位
            constexpr int DECIMAL = 10; // 十进制常量
            auto riter = A.crbegin();
            auto rend = A.crend();
            // 注意循环条件,只有有一个没有被加数没有被使用完成,就应该继续相加
            while (riter != rend || K != 0) {
                int a = riter != rend ? *riter++ : 0; // 如果A循环没有完成取当前位数,如果完成取0
                int sum = K % DECIMAL + a + carry;
                K = K / DECIMAL;
                res.emplace_back(sum % DECIMAL);
                carry = sum / DECIMAL;
            }
            // 累加完成后需要判断最后是否有进位
            if (carry > 0) {
                res.emplace_back(carry);
            }
            return vector<int>(res.crbegin(), res.crend());
        }
    };

 

上一篇:金额最多只能输入两位小数


下一篇:1680. Concatenation of Consecutive Binary Numbers (M)