[LeetCode] Daily Challenge: Add Two Numbers (Python)

LeetCode - Daily Challenge - Add Two Numbers

Problem Description

You are given two non-empty linked lists representing two non-negative integers.

The digits are stored in reverse order, and each of their nodes contains a single digit.

Add the two numbers and return the sum as a linked list.

 

You may assume the two numbers do not contain any leading zero, except the number 0 itself.

 

Example 1 :

• Input: l1 = [2,4,3], l2 = [5,6,4]
• Output: [7,0,8]
• Explanation: 342 + 465 = 807.

Example 2:

• Input: l1 = [0], l2 = [0]
• Output: [0]

Example 3:

• Input: l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
• Output: [8,9,9,9,0,0,0,1]

 

Constraints:

• The number of nodes in each linked list is in the range [1, 100].
• 0 <= Node.val <= 9
• It is guaranteed that the list represents a number that does not have leading zeros.

 

My Solution (Python)

# 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
        """
        result = ListNode()
        answer = result # head of result
        remainder = 0
        sum = 0
        while True:
            if l1 == None and l2 == None:
                if remainder == 1:
                    sum = 1
                    remainder = 0
                else:
                    break
            elif l1 == None and l2 != None:
                sum = l2.val + remainder
                l2 = l2.next
            elif l1 != None and l2 == None:
                sum = l1.val + remainder
                l1 = l1.next
            else:
                sum = l1.val + l2.val + remainder
                l1 = l1.next
                l2 = l2.next
            
            if (sum < 10):
                remainder = 0
            else:
                sum = sum - 10
                remainder = 1

            result.next = ListNode(sum)
            result = result.next
        return answer.next

1. 리턴할 노드의 헤드를 만들어준다. (이는 제출용을 위해 answer로 따로 저장해주었음.) 
2.  반복문: 노드가 Tail인지 확인하기 위한 if elif ... else 문을 이용했다.각 노드의 값이 sum에 더하고 난 뒤에는 다음 노드의 값을 기존의 변수에 저장한다.
   
   1. 두 노드가 Tail일때, 남아있는 나머지를 확인하고 나머지가 없다면 반복문을 종료한다.
   2. 두 노드 중 하나만 Tail일때, Tail이 아닌 노드의 값과 나머지를 더한 sum 값을 구한다.
   3. 두 노드 다 Tail이 아니일때, 두 노드의 값과 나머지를 더한 sum 값을 구한다.
3.  각 노드는 1 digit number만 저장할 수 있으므로 10이 되면 일의 자리 수만 저장하고 다음 노드의 값에 1을 추가하므로 if statement를 이용해 확인해준다.
   1. sum 값이 10 미만일 경우 remainder는 0가 된다.
   2. sum 값이 10 이상일 경우 sum 에서 10을 뺀 뒤  일의 자리 값만 저장, remainder에 1을 저장한다.
4. 반복문이 종료되면 Head인 answer의 다음 값인 answer.next 값을 리턴한다. (answer.val 값이 0이므로)