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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路一：K与数组A的最后一位相加，然后K取余，作为A的最后一个数，然后K除以10，再与A的倒数第二个相加，然后K再取余，作为A的倒数第二个数，K除以10........如果K先到0，则返回结果，如果K特别大，把K剩余的数，采用头插法，插入到数组的头部。时间复杂度O(n)，空间复杂度O(1)

class Solution {
public:
    vector<int> addToArrayForm(vector<int>& A, int K) {
        for(int i = A.size()-1; i >= 0; --i)
        {
            K = A[i] + K;
            A[i] = K%10;
            K /= 10;
            if(K == 0)
                return A;
        }
        vector<int>::iterator it = A.begin();
        while(K > 0)
        {
            A.insert(it, K%10);
            K /= 10;
            it = A.begin();
        }
        return A;
    }
};

思路二：想偷个懒，把数组先换成数，然后两个数相加，再把和变为数组，结果，测试样例中，有一个及其长的数组，就没办法了。

class Solution {
public:
    vector<int> addToArrayForm(vector<int>& A, int K) {
        long long temp = 0;
        long long k = 1;
        for(int i = A.size()-1; i >= 0; --i, k *= 10)
        {
            temp += (long long)(A[i]*k);
        }
        long long sum = temp + K;
        vector<int> res;
        res.push_back(sum%10);
        sum /= 10;
        vector<int>::iterator it = res.begin();
        while(sum > 0)
        {
            res.insert(it, sum%10);
            sum /= 10;
            it = res.begin();
        }
        return res;
    }
};