【刷演算法】LeetCode.2-兩數相加

亞古發表於2019-01-13

給出兩個 非空 的連結串列用來表示兩個非負的整數。其中,它們各自的位數是按照 逆序 的方式儲存的,並且它們的每個節點只能儲存 一位 數字。

如果,我們將這兩個數相加起來,則會返回一個新的連結串列來表示它們的和。

您可以假設除了數字 0 之外,這兩個數都不會以 0 開頭。

示例:

輸入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
輸出:7 -> 0 -> 8
原因:342 + 465 = 807
複製程式碼
/**
 * Definition for singly-linked list.
 * function ListNode(val) {
 *     this.val = val;
 *     this.next = null;
 * }
 */
/**
 * @param {ListNode} l1
 * @param {ListNode} l2
 * @return {ListNode}
 */
var addTwoNumbers = function(l1, l2) {
  if(l1 === null) return l2;
  if(l2 === null) return l1;
  
  let cur1 = l1, cur2 = l2;
  let isCarried =  false;
  let cur = null, head = null;
  
  while(cur1 !== null && cur2 !== null) {
    let sum = cur1.val +  cur2.val;
    sum = isCarried ? sum+1 : sum;
    
    isCarried = false;
    
    if(sum >= 10) {
      sum  = sum % 10;
      isCarried = true;
    }
    
    let node = new ListNode(sum);
    if(head === null) {
      head = node;
    }
    if(cur === null) {
      cur = node;
    } else {
      cur.next = node;
      cur = node;
    }
    cur1 = cur1.next;
    cur2 = cur2.next;
  }
  
  let lastCur = cur1 === null ? cur2 : cur1;
  while(lastCur !== null) {
    let sum = isCarried ? lastCur.val+1 : lastCur.val;
    isCarried = false;
    
    if(sum >= 10) {
      sum = sum % 10;
      isCarried = true;
    }
    
    cur.next = new ListNode(sum);
    cur = cur.next;
    
    lastCur = lastCur.next;
  }
  
  if(isCarried) {
    cur.next = new ListNode(1)
  }
  
  return head;
};
複製程式碼

相關文章