给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
链接:https://leetcode.com/problems/add-two-numbers/
解题思路:
对两个链表同时从头遍历,对应位上的数相加,用l1存储最终结果,判断是否产生进位,将进位加到下一位上。若两链表长短不一,则将长的链表接到l1后面。若最高位仍有进位,则新建结点存储最高位。
我的代码:
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution(object):
def addTwoNumbers(self, l1, l2):
"""
:type l1: ListNode
:type l2: ListNode
:rtype: ListNode
"""
h1 = l1
h2 = l2
pre1 = h1
pre2 = h2
cur = 0
while h1 and h2:
h1.val += h2.val + cur
if h1.val >= 10:
cur = 1
h1.val -= 10
else:
cur = 0
pre1 = h1
pre2 = h2
h1 = h1.next
h2 = h2.next
# 长短不一
if h2:
pre1.next = h2
h1 = h2
while h1:
h1.val += cur
if h1.val >= 10:
cur = 1
h1.val -= 10
else:
cur = 0
break
pre1 = h1
h1 = h1.next
if cur:
pre1.next = ListNode(cur)
return l1
Time: O(max(len(l1), len(l2)))
Space: O(1)