算法学习100天——11 链表 两数之和 归并思想

题目地址(2. 两数相加)

https://leetcode-cn.com/problems/add-two-numbers/

题目描述

给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的，并且每个节点只能存储 一位 数字。
请你将两个数相加，并以相同形式返回一个表示和的链表。
你可以假设除了数字 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]

提示：
每个链表中的节点数在范围 [1, 100] 内
0 <= Node.val <= 9
题目数据保证列表表示的数字不含前导零

思路

本题难道不大，我觉得读题有点难，enmmm，理解题意之后，做这个题很简单，就是归并排序的归并思想

遍历两个链表，每一个结点相加，如果要进位，就把flag标记为1，用完之后立马标记为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 static ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode head = new ListNode();
        ListNode temp = head;
        int flag = 0;
        while(l1 != null && l2 != null){
            // 两数相加 再加上 进位
            int sum = l1.val + l2.val + flag;
            // 进位归零
            flag = 0;
            // 判断这次有没有进位
            if(sum >= 10){
                sum = sum % 10;
                flag = 1;
            }
            temp.next = new ListNode(sum);
            temp = temp.next;
            l1 = l1.next;
            l2 = l2.next;
        }

        // 谁没加完 继续
        while (l1 != null){
            int sum = l1.val + flag;
            // 进位归零
            flag = 0;
            // 判断这次有没有进位
            if(sum >= 10){
                sum = sum % 10;
                flag = 1;
            }
            temp.next = new ListNode(sum);
            temp = temp.next;
            l1 = l1.next;
        }
        while (l2 != null){
            int sum = l2.val + flag;
            // 进位归零
            flag = 0;
            // 判断这次有没有进位
            if(sum >= 10){
                sum = sum % 10;
                flag = 1;
            }
            temp.next = new ListNode(sum);
            temp = temp.next;
            l2 = l2.next;
        }
        
        if (flag > 0){
            temp.next = new ListNode(flag);
        }

        return head.next;
    }
}

结果

执行用时：1 ms, 在所有 Java 提交中击败了100.00%的用户

内存消耗：41.4 MB, 在所有 Java 提交中击败了24.12%的用户