leetcode 989. 数组形式的整数加法(python)

题目链接

题目描述：

对于非负整数 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

解法一：
将数组转换成数字，加上k之后在转换成数组
注意在将数组转换成数字的时候不能用下面这种方法，因为flag最大是10000，求10的一万次方会超时

A=A[::-1]
        num=0
        flag=0
        for i in A:
            num+=i*10**flag
            flag+=1

class Solution:
    def addToArrayForm(self, A: List[int], K: int) -> List[int]:
        num=0
        for i in A:
            num=num*10+i
        num+=K
        res=[]
        if num==0:
            return [0]
        while num!=0:
            res.append(num%10)
            num=num//10
        return res[::-1]

解法二：
将数组各位与K的各位逐位相加，并记录下每一次相加的进位
注意有可能K的位数比数组的位数多

class Solution(object):
    def addToArrayForm(self, A, K):
        """
        :type A: List[int]
        :type K: int
        :rtype: List[int]
        """
        A=A[::-1]
        flag=0#表示进位
        for i in range(len(A)):
            a=K%10
            x=A[i]+a+flag
            flag=x//10
            A[i]=x%10
            K=K//10
        while  K!=0 or flag!=0:
            a=K%10
            x=a+flag
            flag=x//10
            A.append(x%10)
            K=K//10
        return A[::-1]

p.s.评论区答案
用k作为判定条件，当A的长度小于K时，在A后面添加0

        A = A[::-1]
        temp = 0
        i = 0
        while (K!=0):
            if i>(len(A)-1):
                A.append(0)
            temp = K + A[i]
            K = temp/10
            A[i] = temp%10
            i += 1
        return A[::-1]