python - 求两个数组的和

Question

我正在尝试在 Python 中查找两个列表/数组的总和。

例如:

你得到两个随机整数列表，分别是 lst1 和 lst2，大小分别为 n 和 m。这两个列表都包含从 0 到 9 的数字(即，每个索引都存在一位整数)。

这里的想法是将每个列表表示为数字 N 和 M 本身的整数。

您需要找到两个输入列表的总和，将它们视为两个整数并将结果放入另一个列表中，即输出列表也将在每个索引处仅包含单个数字。

以下是我尝试过的代码:

def list_sum(lst1, n, lst2, m) :
    i, j, sum, carry = 0, 0, 0, 0
    new_lst = []
    if n == 0 and m == 0:
        new_lst.append(0)
    elif n > 0 and m>0:
        while n > 0 and m > 0:
            sum = lst1[n - 1] + lst2[m - 1] + carry
            if sum >= 10:
                carry = 1
            else:
                carry = 0
            new_lst.append(sum % 10)
            n -= 1
            m -= 1
        while n > 0:
            if (lst1[n-1] + carry) >= 10:
                new_lst.append((lst1[n-1] + carry) % 10)
                carry = 1
            else:
                new_lst.append(lst1[n-1])
                carry = 0
            n -= 1
        while m > 0:
            if (lst2[m-1] + carry) >= 10:
                new_lst.append((lst2[m-1] + carry) % 10)
                carry = 1
            else:
                new_lst.append(lst1[m-1])
                carry = 0
            m -= 1
        if carry == 1:
            new_lst.append(1)
        new_lst.reverse()
    elif n == 0 and m > 0:
        new_lst.append(0)
        new_lst = new_lst + lst2
    elif n > 0 and m == 0:
        new_lst.append(0)
        new_lst = new_lst + lst1
    print(new_lst)

但是我觉得我在这里遗漏了一些东西，这并没有给我正确的组合答案。有时它错误列出了索引错误。我不知道为什么。

示例输入:

n = 3
lst1 = [6, 9, 8] 
m = 3
lst2 = [5, 9, 2]

输出:

[1, 2, 9, 0]

在这里，每个元素被求和，然后如果 sum >=10 则我们得到一个 carry = 1 并将与下一个总和相加。

即

1. 8+2= 10 >=10 hence carry=1 in first sum
2. 9+9+1( carry) = 19 >=10 hence carry=1
3. 6+5+1( carry) = 12>=10 hence carry=1
4. upend the carry to next position as 1
Hence resultant list would be [1, 2, 9, 0]

接下来我可以尝试什么？

Answer 1

嗯，所有其他答案都非常适合添加 2 个数字(数字列表)。
但是如果你想创建一个可以处理任意数量的“数字”的程序，

Here's what you can do...

def addNums(lst1, lst2, *args):
    numsIters = [iter(num[::-1]) for num in [lst1, lst2] + list(args)]  # make the iterators for each list
    carry, final = 0, []                                                # Initially carry is 0, 'final' will store the result
    
    while True:
        nums = [next(num, None) for num in numsIters]                   # for every num in numIters, get the next element if exists, else None
        if all(nxt is None for nxt in nums): break                      # If all numIters returned None, it means all numbers have exhausted, hence break from the loop
        nums = [(0 if num is None else num) for num in nums]            # Convert all 'None' to '0'
        digit = sum(nums) + carry                                       # Sum up all digits and carry
        final.append(digit % 10)                                        # Insert the 'ones' digit of result into final list
        carry = digit // 10                                             # get the 'tens' digit and update it to carry

    if carry: final.append(carry)                                       # If carry is non-zero, insert it
    return final[::-1]                                                  # return the fully generated final list

print(addNums([6, 9, 8], [5, 9, 2]))                                    # [1, 2, 9, 0]
print(addNums([7, 6, 9, 8, 8], [5, 9, 2], [3, 5, 1, 7, 4]))             # [1, 1, 2, 7, 5, 4]

希望这是有道理的!