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

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

你可能感兴趣的:(989. 数组形式的整数加法(Python))