2、两数相加

image.png
# Definition for singly-linked list.
class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None

class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        prenode = ListNode(0)
        lastnode = prenode
        val = 0
        while val or l1 or l2:
            val, cur = divmod(val + (l1.val if l1 else 0) + (l2.val if l2 else 0), 10)
            lastnode.next = ListNode(cur)
            lastnode = lastnode.next
            l1 = l1.next if l1 else None
            l2 = l2.next if l2 else None
        return prenode.next


def generateList(l: list) -> ListNode:
    prenode = ListNode(0)
    lastnode = prenode
    for val in l:
        lastnode.next = ListNode(val)
        lastnode = lastnode.next
    return prenode.next

def printList(l: ListNode):
    while l:
        print("%d, " %(l.val), end = '')
        l = l.next
    print('')

if __name__ == "__main__":
    l1 = generateList([1, 5, 8])
    l2 = generateList([9, 1, 2, 9])
    printList(l1)
    printList(l2)
    s = Solution()
    sum = s.addTwoNumbers(l1, l2)
    printList(sum)

呐这个题目的用意是处理超大型数字相加出现溢出的问题,题目涉及了Python链表、数据结构的知识,解法也是我参考别人的(捂脸),但是真的很巧妙。
程序的精髓是Solution类中的addTwoNumbers()函数,核心思想就是从个位开始按位相加元素,然后有进位机制。divmod()是Python内置函数,传入两个整数,返回值是其其相除结果(取整)以及余数,用在这里恰到好处:遇十进一。
最后可以测试一下结果:851+9219=10070,allright,完整代码就放在这里,还有不懂的地方可以单步调试。

你可能感兴趣的:(2、两数相加)