Python/알고리즘

[파이썬 알고리즘 인터뷰] 16. 두 수의 덧셈

AI 꿈나무 2021. 2. 19. 15:20
반응형

 

 이 포스팅은 파이썬 알고리즘 인터뷰를 공부하면서 정리한 내용입니다.

 

 출처 : 파이썬 알고리즘 인터뷰

 코드 출처 : 파이썬 알고리즘 인터뷰 깃허브

 


16. 두 수의 덧셈

리트코드 2. Add Two Numbers 문제입니다.

 

풀이

def addTwoNumbers(self, l1, l2):
    root = head = ListNode(0)
    
    carry = 0
    
    while l1 or l2 or carry:
        sum = 0
        if l1:
            sum += l1.val
            l1 = l1.next
        if l2:
            sum += l2.val
            l2 = l2.next
        
        carry, val = divmod(sum + carry,10)
        head.next = ListNode(val)
        head = head.next
    return root.next
반응형