-
Notifications
You must be signed in to change notification settings - Fork 7
/
add_two_numbers.py
43 lines (36 loc) · 1.04 KB
/
add_two_numbers.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def addTwoNumbers(self, l1, l2):
"""
:type l1: ListNode
:type l2: ListNode
:rtype: ListNode
"""
l3_head = ListNode(0)
l3 = l3_head
carry = 0
while l1 != None and l2 != None:
l3.next = ListNode(l1.val + l2.val + carry)
carry = l3.next.val // 10
l3.next.val -= carry*10
l1 = l1.next
l2 = l2.next
l3 = l3.next
if l1 == None:
temp = l2
else:
temp = l1
while temp != None:
l3.next = ListNode( temp.val + carry )
carry = l3.next.val // 10
l3.next.val -= carry*10
temp = temp.next
l3 = l3.next
if carry > 0:
l3.next = ListNode(carry)
l3_head = l3_head.next
return l3_head