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

代码实现：
C++:

class Solution {
public:
    vector<int> addToArrayForm(vector<int>& A, int K) {
        // 输入的数字不可能大于10位数，+1表示有可能需要进位
        int len = A.size()>10? A.size()+1:11;
        vector<int> res(len, 0);
        int i = A.size() - 1;
        int j = len -1;
        for(; i>=0; i--, j--){
            // 数组的从最后一位开始，都与K的每一位相加
            K = A[i] + K;
            res[j] = K%10;
            // 将数字K去掉末尾
            K = (int)(K/10);
        }
        //数组扫描完之后，判断K还有数字吗，
        //其实就是每一位对应相加之后，K的位数是否比数组大小要大
        for(; K>0; K=(int)(K/10), j--)
        {
            res[j] = K%10;
        }
        //整理结果
        vector<int> result;
        for(int i=j+1; i<len; i++)
            result.push_back(res[i]);
        return result;
    }
};

Python：

class Solution:
    def addToArrayForm(self, A, K):
        res = []
        for Figure in A[::-1]:
            K = K + Figure
            res.append(K%10)
            K = K//10
        
        # 判断K的位数是否大于数组A的大小
        while K:
            res.append(K%10)
            K = K//10
        
        return res[::-1]