Pythonで配列形式の整数に数値を加算する方法

ある整数が配列（リスト）形式で与えられている状況を考えてみましょう。例えば、数値が534の場合、[5, 3, 4] のように各桁が要素として格納されています。この配列形式の数値に対して、別の値 k を加算し、結果も同様に桁ごとの配列として返すのが課題です。

解法のアプローチ

この問題は、以下の手順で解くことができます。

• 配列内の各桁を文字列に変換し、それらを連結して1つの文字列を作る
• 連結した文字列を整数に変換し、k を加算する
• 計算結果を再び文字列に変換し、各桁を取り出して配列（リスト）を作成する

実装例

以下のコードは、上記の手順を実装したものです。動作をより深く理解するために参考にしてください。

class Solution(object):
    def addToArrayForm(self, A, K):
        num_A = int(''.join(str(i) for i in A))
        res = list(str(num_A+K))
        res = list(map(int,res))
        return res
ob1 = Solution()
print(ob1.addToArrayForm([5,3,4], 78))

入力

[5,3,4]
78

出力

[6,1,2]

コードの解説

まず ''.join(str(i) for i in A) によって、配列 [5, 3, 4] の各要素を文字列化して連結し、「534」という文字列を生成します。次に int() で整数に変換し、k（ここでは78）を加算することで 612 が得られます。最後に結果を文字列化して list() と map(int, ...) を使うことで、各桁を整数要素とするリスト [6, 1, 2] が完成します。

なお、この手法はシンプルで直感的ですが、非常に大きな桁数の数値を扱う場合は、桁ごとに繰り上がりを処理する方式の方が効率的な場合もあります。用途に応じて使い分けるとよいでしょう。