美文网首页
LeetCode 445. 两数相加 II

LeetCode 445. 两数相加 II

作者: 大梦三千秋 | 来源:发表于2020-04-14 20:13 被阅读0次

    445. 两数相加 II


    题目来源:https://leetcode-cn.com/problems/add-two-numbers-ii

    题目


    给你两个 非空 链表来代表两个非负整数。数字最高位位于链表开始位置。它们的每个节点只存储一位数字。将这两数相加会返回一个新的链表。

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

    进阶:

    如果输入链表不能修改该如何处理?换句话说,你不能对列表中的节点进行翻转。

    示例:

    输入:(7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4)
    输出:7 -> 8 -> 0 -> 7
    

    解题思路


    思路:栈

    在进阶提示中,要求不对输入链表进行修改,不能对链表中的节点进行翻转。

    那么这里就可以考虑使用栈,利用栈先入后出的特性逆序处理链表。

    在处理栈元素返回链表时,使用头插法处理,确保返回的链表顺序正确。

    具体过程可以看下图:

    图解

    代码实现


    # 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:
            # 入栈方法
            def push_stack(node, stack):
                while node:
                    stack.append(node.val)
                    node = node.next
            # 定义栈
            stack1 = []
            stack2 = []
    
            # 执行入栈操作
            push_stack(l1, stack1)
            push_stack(l2, stack2)
    
            # 记录进位
            carry = 0
            # 哑节点,控制链表返回
            dummy = ListNode(0)
    
            # 进行计算
            # 终止条件:当栈为空,或者进位为 0
            while stack1 or stack2 or carry:
                a, b = 0, 0
                # 出栈计算
                if stack1:
                    a = stack1.pop()
                if stack2:
                    b = stack2.pop()
                sum_ = a + b + carry
                carry = sum_ // 10
                mod = sum_ % 10
    
                # 使用头插法
                # 将结果放入返回链表中
                # 这样返回的链表顺序才是正确的
                node = ListNode(mod)
                node.next = dummy.next
                dummy.next = node
    
            return dummy.next
    
    

    实现结果


    实现结果

    以上就是使用栈先入后出的特性,来逆序处理链表,进而解决《445. 两数相加 II》问题的主要内容。其中返回新链表时,使用头插法,确保返回的链表顺序正确。


    欢迎关注微信公众号《书所集录》

    相关文章

      网友评论

          本文标题:LeetCode 445. 两数相加 II

          本文链接:https://www.haomeiwen.com/subject/ywrxvhtx.html