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]