【Leetcode】Python实现两数相加(附上测试链表)

给定两个非空链表来表示两个非负整数。位数按照逆序方式存储,它们的每个节点只存储单个数字。将两数相加返回一个新的链表。

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

示例:

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

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


class Solution(object):
    def addTwoNumbers(self, l1, l2):
        """
        :type l1: ListNode
        :type l2: ListNode
        :rtype: ListNode
        """
        # 这里一开始不做l1、l2非空判断,题意表明非空链表
        # 记录是否需要增加新节点,或在链表下一个节点是否需要加1,同时记录链表同级节点的和
        carry = 0
        # 这里的执行顺序是res = ListNode(0), pre = res
        res = pre = ListNode(0)
        # 判断l1、l2、carry是否有值,carry有值的话需要增加新节点,或在链表下一个节点需要加1
        while l1 or l2 or carry:
            # 判断l1是否有值
            if l1:
                carry += l1.val
                l1 = l1.next
            # 判断l2是否有值
            if l2:
                carry += l2.val
                l2 = l2.next
            # carry有同级节点的和
            # divmod返回商与余数的元组,拆包为carry和val
            # carry有值的话需要增加新节点,或在链表下一个节点需要加1,在循环中会用到
            carry, val = divmod(carry, 10)
            # 新建链表节点
            # 这里是n.next = ListNode(val), n = n.next()
            pre .next = pre  = ListNode(val)
        # res等价于pre,res.val=0,所以返回res.next
        return res.next


if __name__ == '__main__':
    # 创建对象Solution
    sol = Solution()
    # 定义l1链表
    l1 = ListNode(2)
    l1.next = l11 = ListNode(4)
    l11.next = l12 = ListNode(5)
    # 定义l2链表
    l2 = ListNode(5)
    l2.next = l21 = ListNode(6)
    l21.next = l22 = ListNode(4)
    # 获取返回值的链表
    res = sol.addTwoNumbers(l1, l2)
    # 循环遍历出来
    while res:
        print(res.val)
        res = res.next
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/chenhua1125/article/details/80339751
个人分类: Python Leetcode
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

【Leetcode】Python实现两数相加(附上测试链表)

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭