2. Add Two Numbers

[zeikar]

Problem link

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

Problem Summary

링크드 리스트 형태로 수가 2개 주어질 때 더한 값을 링크드 리스트로 출력하는 문제. 자리수는 계산하기 쉽게 반대로 되어 있다.

Solution

간단한 링크드 리스트 구현 문제. 그냥 자리수대로 한 칸 씩 앞으로 가면서 더해주면 된다.

Source Code

# 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: Optional[ListNode], l2: Optional[ListNode]) -> Optional[ListNode]:
        p1 = l1
        p2 = l2

        head = ListNode()
        pr, r = None, head
        carry = 0

        while p1 != None or p2 != None:
            v1, v2 = 0, 0
            if p1 != None:
                v1 = p1.val
                p1 = p1.next
            if p2 != None:
                v2 = p2.val
                p2 = p2.next
            
            r.val = (v1 + v2 + carry) % 10
            carry = (v1 + v2 + carry) // 10

            pr = r
            r.next = ListNode()
            r = r.next
        
        if carry != 0:
            r.val = 1
        else:
            pr.next = None
            
        return head