動機

要記得進位

Problem

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.

Sol

主要就是進位,一直走下去 要注意是最後還有沒有進位,如果有要多一個node

def size(l):
    ret = 0
    while l:
        l = l.next
        ret += 1
    return ret

class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        car = 0
        # premise: l1.len > l2.len
        if size(l2) > size(l1):
            l1, l2 = [l2, l1]
        ret = l1
        prev = l1
        while l1:
            tmp = l1.val + car + (l2.val if l2 else 0)
            l1.val = tmp % 10
            car = tmp // 10
            
            prev = l1
            l1 = l1.next if l1 else None
            l2 = l2.next if l2 else None
        if car != 0:
            prev.next = ListNode(car)
        return ret