2.两数相加

Posted 2020-11-25 lzqayx

题目：
给出两个?非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照?逆序?的方式存储的，并且它们的每个节点只能存储?一位?数字。

如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。
链接：https://leetcode-cn.com/problems/add-two-numbers

就是学习了python divmod这个函数

# 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:
        toadd = 0
        head = ListNode(0)
        tail = head        
        while l1 or l2:
            if not l1:
                toadd,a = divmod(l2.val+toadd,10)
            elif not l2:
                toadd,a = divmod(l1.val+toadd,10)
            else:
                toadd,a = divmod(l1.val+l2.val+toadd,10)
            tail.next = ListNode(a)
            tail = tail.next
            if l1:
                l1 = l1.next
            if l2:
                l2 = l2.next
        if toadd>0:
            tail.next = ListNode(toadd)
        return head.next # 初始为了方便头部放了一个0