LEETCODE | PYTHON | 989 | 数组形式的整数加法

LEETCODE | PYTHON | 989 | 数组形式的整数加法

1. 题目

整数的 数组形式  num 是按照从左到右的顺序表示其数字的数组。

例如，对于 num = 1321 ，数组形式是 [1,3,2,1] 。
给定 num ，整数的 数组形式 ，和整数 k ，返回 整数 num + k 的 数组形式 。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/add-to-array-form-of-integer
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 代码

class Solution:
    def addToArrayForm(self, num: List[int], k: int) -> List[int]:

        #num = [0]
        #k = 10000

        #初始化
        l = len(num)
        flag = 0        #计数器：是否进位
        index = l-1     #指针：指向List的位置

        #从List末尾进行相加
        while k>0 and index>=0:

            #计算当前位置的加法结果：List原数值+k%10+进位
            tmp = num[index] + k%10

            #填写List新结果
            num[index] = tmp%10 

            #计算是否需要进位
            flag = tmp//10

            #print(num[index],flag,k)

            #更新k值和index值
            k = k//10 + flag
            index = index - 1

            #print(num[index],flag,k)
        
        #若最后一次计算需进位
        if k == 0 and flag == 1:
            num.insert(0,1)
        
        #若k值仍存在超过num的位数
        while k>0:
            num.insert(0,k%10)
            k = k//10

        return num