两条链表相同位数相加[中等]

优质博文IT-BLOG-CN

一、题目

给你两个非空的链表，表示两个非负的整数。它们每位数字都是按照逆序的方式存储的，并且每个节点只能存储一位数字。请你将两个数相加，并以相同形式返回一个表示和的链表。你可以假设除了数字0之外，这两个数都不会以0开头。

示例 1：
输入：l1 = [2,4,3], l2 = [5,6,4]
输出：[7,0,8]
解释：342 + 465 = 807.

示例 2：
输入：l1 = [0], l2 = [0]
输出：[0]

示例 3：
输入：l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
输出：[8,9,9,9,0,0,0,1]

二、代码

/**
 * 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 current = head;
        
        boolean isTen = false;
        while (l1 != null || l2 != null) {
            int l1Value = l1 == null ? 0 : l1.val;
            int l2Value = l2 == null ? 0 : l2.val;
            // 注意：如果上一次大于10的时候，不要忘记+1
            int total = isTen ? l1Value + l2Value + 1 : l1Value + l2Value;
            if (head == null) {
                head = new ListNode(total %10);
                current = head;
            } else {
                // 注意：给当前节点的next赋值完成后，立即让当前节点指向tail节点
                current.next = new ListNode(total %10);
                current = current.next;
            }
            isTen = total >= 10;
            // 注意：重新定义l1和l2不然会死循环
            l1 = l1 != null ? l1.next : null;
            l2 = l2 != null ? l2.next : null;
        }
        // 注意：大于10的时候，还要在tail中添加一个节点
        if (isTen) {
            current.next = new ListNode(1);
        }
        return head;
    }
}