leetcode 2 python 两数相加

传送门

题目要求

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

如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。

您可以假设除了数字 0 之外，这两个数都不会以 0 开头。

示例：
输入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出：7 -> 0 -> 8
原因：342 + 465 = 807

思路

遍历2个链表，相当于从右到左进行2个整数的遍历，直接将其值相加作为新的节点值放入新链表，注意当值>=10的时候有进位；注意2个链表长度不一定相同（一个耗尽另一个还要继续遍历）；注意最后得到的结果是否有进位。比如（1234，976599），虽然链表1先耗尽，但最后结果仍然会有进位，记得要判断一下

talk is cheap, show me the code

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

class Solution:
    def addTwoNumbers(self, l1, l2):
        """
        :type l1: ListNode
        :type l2: ListNode
        :rtype: ListNode
        """
        p, q = l1, l2
        # 一个是头结点，一个作为指针移动
        m = n = ListNode(0)
        # 记录进位，初始值为0（因为是整数，个位用不上）
        up = 0
        while p and q:
            # 新链表的下一个节点val
            s = p.val + q.val + up
            # 看是否有进位，有的话先用变量up保存，下一位使用
            up = 1 if s >= 10 else 0
            # 新链表的下一节点
            n.next = ListNode(s % 10)
            p, q, n = p.next, q.next, n.next
        # 同时遍历后剩下的链表
        l = p or q
        # 如法炮制
        while l:
            s = l.val + up
            up = 1 if s >= 10 else 0
            n.next = ListNode(s % 10)
            l, n = l.next, n.next
        # 最后要再判断一下是否有进位，有的话要带上
        if up:
            n.next = ListNode(1)
        return m.next