Leetcode之两数相加问题

Leetcode之两数相加问题（python）

---

给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的，并且每个节点只能存储 一位 数字。

请你将两个数相加，并以相同形式返回一个表示和的链表。

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

1. 代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        head=cur=ListNode(0)
        #用于进位
        a=0
        #循环两个链表
        while l1 or l2:
        	#指向第一个链表l1，并更新当前值b，以及进位a
            cur.next = l1
            b=l1.val + l2.val + a
            if b>=10:
                b=b-10
                a=1
            else:
                a=0
            l1.val=b
			
			#如果两个链表next都不为空，直接赋值
            if l1.next and l2.next:
                l1=l1.next
                l2=l2.next
            #假设至少有一个链表next为空
            else:
                #如果l1.next不是None，则l2.next一定是None，加长L2用0补位。
                if l1.next:
                    l1=l1.next
                    l2=ListNode(0)
                #如果l2.next不是None，则l1.next一定是None，加长L1用0补位。
                elif l2.next:
                    l1=ListNode(0)
                    l2=l2.next
                #如果l2.next和l1.next不是None。
                else:
                	#判断最后一位是否还有进位，若有a>0则继续，否则停止。
                    if a==0:
                        break
                    l1=ListNode(0)
                    l2=ListNode(0)
            cur = cur.next
        cur.next = l1
        return head.next

代码有些过于繁琐，想了半天也只能按照一个个判断分叉写下去，内存消耗太大……