LeetCode_989_数组形式的整数加法 | 廖家龙的博客网站

题目描述：

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

示例：

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

输入：A = [2,7,4], K = 181
输出：[4,5,5]
解释：274 + 181 = 455

输入：A = [2,1,5], K = 806
输出：[1,0,2,1]
解释：215 + 806 = 1021

输入：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. 1 <= A.length <= 10000
2. 0 <= A[i] <= 9
3. 0 <= K <= 10000
4. 如果 A.length > 1，那么 A[0] != 0

解法1：逐项求和

class Solution {
public:
    vector<int> addToArrayForm(vector<int>& A, int K) {

        //输入：A = [1,2,3], K = 912
        //输出：[1,0,3,5]
        vector<int> res;

        //carry为1表示需要进位
        int i = A.size() - 1, carry = 0;

        while (i >= 0 || K != 0) {

            int x = i >= 0 ? A[i] : 0;  //x每次取数组A最后一位
            int y = K != 0 ? K % 10 : 0;  //y每次取K最后一位

            int sum = x + y + carry;

            res.push_back(sum % 10);

            carry = sum / 10;

            i--;
            K = K / 10;
        }

        if (carry != 0) res.push_back(carry);

        reverse(res.begin(), res.end());

        return res;
    }
};