更多精彩內(nèi)容，請關(guān)注【力扣簡單題】。

題目

難度：★☆☆☆☆
類型：數(shù)組

對于非負整數(shù) X 而言，X 的數(shù)組形式是每位數(shù)字按從左到右的順序形成的數(shù)組。例如，如果 X = 1231，那么其數(shù)組形式為 [1,2,3,1]。

給定非負整數(shù) X 的數(shù)組形式 A，返回整數(shù) X+K 的數(shù)組形式。

提示
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：轉(zhuǎn)為數(shù)字進行相加

python中有多種內(nèi)置函數(shù)可以幫助我們實現(xiàn)快速計算：

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

轉(zhuǎn)為字符串列表map(str, A)：['1', '2', '0', '0']

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

字符串轉(zhuǎn)數(shù)字并相加int(''.join(map(str, A)))+K：1200+34=1234

相加結(jié)果轉(zhuǎn)成數(shù)字列表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

如有疑問或建議，歡迎評論區(qū)留言~