Skip to content

Latest commit

 

History

History
75 lines (62 loc) · 2.3 KB

2. 两数相加-Medium.md

File metadata and controls

75 lines (62 loc) · 2.3 KB

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

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

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

示例:

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

Solution

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:
        start = None
        flag = 0

        while flag or l1 or l2:
            tmp_sum = 0
			tmp_sum = (l1.val if l1 else 0) + (l2.val if l2 else 0) + flag
			if l1: l1 = l1.next
			if l2: l2 = l2.next
			
			flag = tmp_sum // 10
			tmp_sum %= 10

			tmpNode = ListNode(tmp_sum)			
            if not start:
                start = tmpNode
            else:
                current.next = tmpNode
			current = tmpNode

        return start

2. 精简代码(递归写法,from 题解)

思路

根据题意可知链表数字位数是从小到大的

  1. 因为两个数字相加会产生进位,所以使用i来保存进位。
  2. 则当前位的值为(l1.val + l2.val + i) % 10
  3. 则进位值为(l1.val + l2.val + i) / 10
  4. 建立新node,然后将进位传入下一层。
# 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:
        def dfs(l1, l2, flag):
            if not l1 and not l2 and not flag:
                return None
            tmp_sum = (l1.val if l1 else 0) + (l2.val if l2 else 0) + flag
            node = ListNode(tmp_sum % 10)
            node.next = dfs(l1.next if l1 else l1, l2.next if l2 else l2, tmp_sum // 10)
            return node
			
        return dfs(l1, l2, 0)