leetcode - 2.两数相加(链表)

367 阅读1分钟

给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。

如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。 您可以假设除了数字 0 之外,这两个数都不会以 0 开头。

示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
  • 就像做数学加法,从后往前加,两数相加就进位(carry:进位标志)
  • 当某一条链表已经提前遍历完,继续遍历直到最长的链表遍历完,这时候短的链表为空的节点相加时默认为0
  • 结束的条件是两条链表都已经遍历完成
  • 要考虑的几种条件:
    • 一条链表遍历完了,一条没完(遍历完的链表后续所有节点的值都以0存在)
    • 全部遍历完成(l1、l2都为空)
    • 全部遍历完成,还有进位(l1 ==null && l2 == null && carry > 0)
/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
      ListNode head = null;
      ListNode tail = null;
      int carry = 0;
      
      while(l1 != null || l2 != null) {
        int a = l1 == null ? 0 : l1.val;
        int b = l2 == null ? 0 : l2.val;
        int sum = a + b + carry;
        
        carry = sum / 10;
        if(head == null) {
          head = tail = new ListNode(sum % 10);
        } else {
          tail.next = new ListNode(sum % 10);
          tail = tail.next;
        }
        
        if(l1 != null) {
          l1 = l1.next;
        }
        if(l2 != null) {
          l2 = l2.next;
        }
      }
      if(carry > 0) {
        tail.next = new ListNode(carry); 
      }
      return head;
    }
}