LeetCode 2:两数相加(自己写的)

给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。

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

思路:
其实这道题主要是为了考察我们对链表的熟练使用,算法很简单,逆序已经使得计算过程简单了很多,链表从左到右不就是个十百千位吗,两个指针直接从两个链表的开始直接遍历就可以了。停止条件是什么,当然是两个给定链表都被遍历完为止,即两个指针均为None。
接下来考虑一些特殊情况:
1、对应位数字相加大于10要进位,所以要设置一个变量来存储进位数。
2、两个链表长度不一致,继续遍历就会报错了,因为短链表指针为None值了啊。此时我们只需要给短的链表后面挂上一个值为0的结点就可以了继续了。就好象下图:
LeetCode 2:两数相加(自己写的)_第1张图片
3、最高位相加大于十,也要进位,这个肯定不能加在循环里,只能在最后判断,两个给定链表都遍历完了,如果存储进位数的变量不等于0,就需要在存储结果的链表中再挂一个节点,值设置为1也就是进位变量就可以了。
4、最后记得要加头指针便于输出。
代码如下:

class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None


class Solution:
    def addTwoNumbers(self, l1, l2):
        # L = ListNode(x=0)
        l = ListNode(x=0)
        prenode = ListNode(x=0)
        L=prenode
        a = 0
        while l1 or l2:
            if not l1: l1 = l
            if not l2: l2 = l
            L.next = ListNode((l1.val + l2.val + a)%10)
            a=(l1.val + l2.val + a)//10
            l1 = l1.next
            l2 = l2.next
            L = L.next
        if a>0:
            L.next=ListNode(a)
        return prenode.next


def printlist(l:ListNode)->ListNode:#输出结果链表的函数,提交的时候不需要写,用来测试的。
    while l:
        print(l.val)
        l=l.next

你可能感兴趣的:(LeeCode)