Java教程

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

本文主要是介绍算法学习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%的用户

这篇关于算法学习100天——11 链表 两数之和 归并思想的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!