leetcode-python/add_linked_lists.py at master · ashishmh/leetcode-python · GitHub

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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import sys

# https://leetcode.com/problems/add-two-numbers/

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


def addTwoNumbers(l1, l2):
	if l1 is None:
        return l2
    if l2 is None:
        return l1

    resultPtr = None
    result = None
    carry = 0
    while (l1 is not None and l2 is not None):
        sum = l1.val + l2.val + carry
        if sum > 9:
            carry = 1
            sum = sum % 10
        else:
            carry = 0
        if resultPtr is None:
            resultPtr = ListNode(sum)
            result = resultPtr
        else:
            resultPtr.next = ListNode(sum)
            resultPtr = resultPtr.next
        l1 = l1.next
        l2 = l2.next
    while l1 is not None:
        sum = l1.val + carry
        if sum > 9:
            carry = 1
            sum = sum % 10
        else:
            carry = 0
        resultPtr.next = ListNode(sum)
        resultPtr = resultPtr.next
        l1 = l1.next
    while l2 is not None:
        sum = l2.val + carry
        if sum > 9:
            carry = 1
            sum = sum % 10
        else:
            carry = 0
        resultPtr.next = ListNode(sum)
        resultPtr = resultPtr.next
        l2 = l2.next
    if carry != 0:
        resultPtr.next = ListNode(carry)

    return result


def main():
	str = input('input: ')
	output = addTwoNumbers()
	print('output: ', output)


if __name__ == '__main__':
    main()