LeetCode 445. 两数相加 II

作者: TheKey_ | 来源:发表于2019-07-05 16:35 被阅读0次

    445. 两数相加 II

    给定两个非空链表来代表两个非负整数。数字最高位位于链表开始位置。它们的每个节点只存储单个数字。将这两数相加会返回一个新的链表。

    你可以假设除了数字 0 之外,这两个数字都不会以零开头。

    进阶:

    如果输入链表不能修改该如何处理?换句话说,你不能对列表中的节点进行翻转。

    示例:
    输入: (7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4)
    输出: 7 -> 8 -> 0 -> 7
    

    来源:力扣(LeetCode)
    链接:https://leetcode-cn.com/problems/add-two-numbers-ii/
    著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


    • 1.使用递归 + 反转

    思路:
    1.将每个链表都进行反转,之后按照之前的方法一次相加, 思路和上篇文章类似(LeetCode 2. 两数相加 - 简书
    3.相加之后在进行反转一次即可

    • 注意 :要小心链表大小不一的情况,获取空链表的变量可能会出现 NullPointerException
    public static class ListNode {
    
            private int val;
            private ListNode next;
    
            public ListNode(int val) {
                this.val = val;
            }
    
            //用于测试用例
            public ListNode(int[] arr) {
                if (arr == null || arr.length == 0) throw new NullPointerException("array is Empty");
                this.val = arr[0];
                ListNode cur = this;
                for (int i = 1; i < arr.length; i++) {
                    cur.next = new ListNode(arr[i]);
                    cur = cur.next;
                }
            }
    
            @Override
            public String toString() {
                StringBuilder res = new StringBuilder();
                ListNode cur = this;
                while (cur != null) {
                    res.append(cur.val + "->");
                    cur = cur.next;
                }
                res.append("NULL");
                return res.toString();
            }
    
        }
    
       public static ListNode addTwoNumbers(ListNode l1, ListNode l2) {
            ListNode reverseL1 = reverse(l1);
            ListNode reverseL2 = reverse(l2);
            return reverse(add(reverseL1, reverseL2, 0));
        }
    
        public static ListNode add(ListNode l1, ListNode l2, int post) {
            if (l1 == null && l2 == null && post == 0) return null;
            int x = l1 == null ? 0 : l1.val;
            int y = l2 == null ? 0 : l2.val;
            int sum = x + y + post;
            ListNode node = new ListNode(sum % 10);
            node.next = add(l1 == null ? null : l1.next, l2 == null ? null : l2.next, sum / 10);
            return node;
        }
        
        public static ListNode reverse(ListNode head) {
            if (head == null || head.next == null) return head;
            ListNode node = reverse(head.next);
            head.next.next = head;
            head.next = null;
            return node;
        }
    

    复杂度分析:
    时间复杂度:O(max(m , n)), m 和 n 表示两个链表的长度
    空间复杂度:O(max(m, n) + ((m + n)+ max(m, n))), 空间复杂度需要将递归的复杂度加上新的节点占用的空间

    • 2. 入栈 + 反转

    思路:

    1. 新建两个栈,并将两个链表的元素一次入栈
    2. 每次出栈栈顶元素相加,步骤同上方法
    3. 最后将新的链表反转一次即可
    public static ListNode addTwoNumbers(ListNode l1, ListNode l2) {
            Stack<ListNode> stack1 = new Stack<>();
            Stack<ListNode> stack2 = new Stack<>();
            ListNode dummyHead = new ListNode(0);
            ListNode prev = dummyHead;
            int mod = 0;
            while (l1 != null) {
                stack1.push(l1);
                l1 = l1.next;
            }
            while (l2 != null) {
                stack2.push(l2);
                l2 = l2.next;
            }
            while (!stack1.isEmpty() || !stack2.isEmpty() || mod != 0) {
                int x = stack1.isEmpty() ? 0 : stack1.peek().val;
                int y = stack2.isEmpty() ? 0 : stack2.peek().val;
                int sum = x + y + mod;
                mod = sum / 10;
                prev.next = new ListNode(sum % 10);
                prev = prev.next;
                if (!stack1.isEmpty()) stack1.pop();
                if (!stack2.isEmpty()) stack2.pop();
            }
            return reverse(dummyHead.next);
        }
    

    复杂度分析:
    时间复杂度:O(max(m , n)), m 和 n 表示两个链表的长度
    空间复杂度:O(max(m, n) + (m + n) + max(m, n)), 空间复杂度需要将栈的复杂度O(m + n)加上新的节点占用的空间O(max(m, n)) 和反转递归占用的O(max(m, n))

    • 3. 进阶(不使用反转)

    思路:
    链表可能存在几种情况:
    1.链表长度相同,不需要进行处理
    2.一个链表为空,直接返回另一个链表即可
    3.链表长度不同,在短的链表前面补0,直到连个链表长度相同

    • 步骤:
      1.将链表补齐后,依次相加,暂时先不要处理进位的情况
      2.从尾节点到头节点依次遍历,遇到需要进位的情况,就将前一个节点加上 mod
      3.返回该链表即可
    image.png
    public static ListNode addTwoNumbers(ListNode l1, ListNode l2) {
             if (l1 == null) return l2;
             if (l2 == null) return l1;
             ListNode cur1 = l1;
             ListNode cur2 = l2;
             ListNode dummyHead = new ListNode(0);
             ListNode prev = dummyHead;
             while (cur1 != null && cur2 != null) {
                 cur1 = cur1.next;
                 cur2 = cur2.next;
             }
             if (cur1 != null) {    //说明l1长度大于l2,将l2前面补齐
                 while (cur1 != null) {
                     prev.next = new ListNode(0);
                     prev = prev.next;
                     cur1 = cur1.next;
                 }
                 //将部位好的链表和短的链表进行拼接
                 prev.next = l2;
                 cur1 = l1;
                 cur2 = dummyHead.next;
             } else {   //l2长度大于l1,将l1前面补齐
                 while (cur2 != null) {
                     prev.next = new ListNode(0);
                     prev = prev.next;
                     cur2 = cur2.next;
                 }
                 prev.next = l1;
                 cur2 = l2;
                 cur1 = dummyHead.next;
             }
    
             //链表进行相加,暂时先不考虑进位情况
            ListNode sumHead = new ListNode(0);
            ListNode prevSum = sumHead;
            while (cur1 != null) {
                prevSum.next = new ListNode(cur1.val + cur2.val);
                cur1 = cur1.next;
                cur2 = cur2.next;
                prevSum = prevSum.next;
            }
    
            int mod = 0;   //余数
            //从尾节点开始,处理进位情况
            while (prevSum != sumHead) {
                int sum = prevSum.val + mod;
                mod = sum / 10;
                prevSum.val = sum % 10;
                //需要找到prevSum之前的一个节点,用于向前遍历新链表
                ListNode front = sumHead;
                while (front.next != prevSum) {
                    front = front.next;
                }
                prevSum = front;
            }
    
            //处理最后还需要进位的情况
            if (mod > 0) {
                ListNode finalNode = new ListNode(mod);
                finalNode.next = sumHead.next;
                sumHead.next = finalNode;
            }
            return sumHead.next;
        }
    

    复杂度分析:
    时间复杂度:O(m + n + max(m , n)), m 和 n 表示两个链表的长度
    空间复杂度:O(max(m, n)), 最多只需要占用max(m, n)用来存储新的节点

    • 测试用例

    int[] arr1 = new int[] {7, 2, 4, 3};
             int[] arr2 = new int[] {5, 6, 4};
             ListNode listNode1 = new ListNode(arr1);
             System.out.println(listNode1);
             ListNode listNode2 = new ListNode(arr2);
             System.out.println(listNode2);
             System.out.println("两数相加:" + addTwoNumbers(listNode1, listNode2));
    
    • 结果

    7->2->4->3->NULL
    5->6->4->NULL
    两数相加:7->8->0->7->NULL
    

    • 源码

    • 我会随时更新新的算法,并尽可能尝试不同解法,如果发现问题请指正
    • Github

    相关文章

      网友评论

        本文标题:LeetCode 445. 两数相加 II

        本文链接:https://www.haomeiwen.com/subject/yevehctx.html