python 输入整数数组_989. 数组形式的整数加法（Python）

更多精彩内容，请关注【力扣简单题】。

题目

难度：★☆☆☆☆

类型：数组

对于非负整数 X 而言，X 的数组形式是每位数字按从左到右的顺序形成的数组。例如，如果 X = 1231，那么其数组形式为 [1,2,3,1]。

给定非负整数 X 的数组形式 A，返回整数 X+K 的数组形式。

提示

1 <= A.length <= 10000

0 <= A[i] <= 9

0 <= K <= 10000

如果 A.length > 1，那么 A[0] != 0

示例

示例 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：转为数字进行相加

python中有多种内置函数可以帮助我们实现快速计算：

输入A：[1, 2, 0, 0]

转为字符串列表map(str, A)：['1', '2', '0', '0']

列表中元素连接''.join(map(str, A))："1200"

字符串转数字并相加int(''.join(map(str, A)))+K：1200+34=1234

相加结果转成数字列表map(int, str(int(''.join(map(str, A)))+K))：[1, 2, 3, 4]

class Solution:

def addToArrayForm(self, A, K):

"""

:param A: List[int]

:param K: int

:return: List[int]

"""

return list(map(int, str(int(''.join(map(str, A)))+K)))

方案2

按照加法运算法则计算即可。

class Solution(object):

def addToArrayForm(self, A, K):

A[-1] += K

carry = 0

for i in reversed(range(len(A))):

carry, A[i] = divmod(A[i], 10)

if i:

A[i-1] += carry

if carry:

A = list(map(int, str(carry))) + A

return A

如有疑问或建议，欢迎评论区留言~