Leetcode 989.数组形式的整数加法（Add to Array-Form of Integer）

Leetcode 989.数组形式的整数加法

1 题目描述（Leetcode题目链接）

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

2 题解

  带个进位，直接算。

class Solution:
    def addToArrayForm(self, A: List[int], K: int) -> List[int]:
        res = []
        i = len(A) - 1
        carry = 0
        while i >= 0 or K:
            a = A[i] if i >= 0 else 0
            b = K%10 if K else 0
            s = a + b + carry
            carry = s//10
            s %= 10
            res.append(s)
            i -= 1
            K //= 10
        return res[::-1] if not carry else [1] + res[::-1]