Skip to content

2. Add Two Numbers

郭耀展 edited this page Mar 14, 2020 · 3 revisions

原题

You are given two non-empty linked lists representing two non-negative integers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.

You may assume the two numbers do not contain any leading zero, except the number 0 itself.

Example:

Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8
Explanation: 342 + 465 = 807.

题意

给你两个代表非负整型数的非空链表。这些数逆序存储在每个包含一个数字的节点中。现在需要将两个整型数相加,并返回响应的链表。

你可以假设这两个整型数不以 0 开头,除非它本身就是 0。

思路

题目的意思很简单,就是迭代两个链表,并在每一次的迭代中,将两个两个链表的节点的数值相加(如果有进位的话,则加上进位),并存到一个新的链表的当前节点中。最后将新的链表返回即可。

以上是两年前,我的解决思路。但听我这么说,好像现在有其它有趣的新想法?

是的,正常加法流程就是这个样。但现在我假设,两个链表,一个很短,一个很长。在短链表遍历完之后,是否能够将长链表直接接到新链表(结果链表)的尾部。这样一来,长链表后面就不需要再遍历了。

由于 **carry(进位)**的存在,"移花接木"变得没那么容易。在 withCarry(ListNode l, int carry) 方法中,如果 carry 不为 0 ,仍然需要遍历长链表进行值相加。直到 carry0 ,结束遍历。

如今,运行时间缩短到 1ms,速度算是过关了,而接下来就是内存。

之前的做法是,建立一个新的链表,用来存储结果。而 l1l2 ,用完就丢,实在可惜。于是乎,我直接选用 l1 作为结果链表。详细实现见代码

代码

  • AddTwoNumbers.java
package name.guolanren._1to100._1to10.p2;

/**
 * @link https://leetcode.com/problems/add-two-numbers/
 * @author guolanren
 */
public class AddTwoNumbers {

    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode sumListNode = l1;
        ListNode currentNode = sumListNode;
        ListNode prevNode = null;
        int carry = 0;
        while (true) {
            // l1 遍历完了,将 l2 加上进位后,接到 sumListNode 尾
            if (l1 == null) {
                prevNode.next = withCarry(l2, carry);
                break;
            }
            // l2 遍历完了,将 l1 加上进位后,接到 sumListNode 尾
            if (l2 == null) {
                prevNode.next = withCarry(l1, carry);
                break;
            }

            // 计算当前节点值
            currentNode.val = l1.val + l2.val + carry;

            // 是否需要进位
            if (currentNode.val > 9) {
                currentNode.val -= 10;
                carry = 1;
            } else {
                carry = 0;
            }

            l1 = l1.next;
            l2 = l2.next;
            prevNode = currentNode;
            currentNode = currentNode.next;
        }
        return sumListNode;
    }

    /**
     * ListNode 加上进位
     */
    private ListNode withCarry(ListNode l, int carry) {
        ListNode head = l;
        if (l != null) {
            while (true) {
                l.val += carry;

                if (l.val < 10) {
                    break;
                }

                l.val -= 10;
                carry = 1;

                if (l.next == null) {
                    l.next = new ListNode(carry);
                    break;
                }

                l = l.next;
            }
        } else if (carry > 0) {
            return new ListNode(carry);
        }
        return head;
    }
}
  • ListNode.java
package name.guolanren._1to100._1to10.p2;

/**
 * @link https://leetcode.com/problems/add-two-numbers/
 * @author guolanren
 */
public class ListNode {
    int val;
    ListNode next;
    ListNode(int x) { val = x; }
}

得意一下

Submission Detail

Clone this wiki locally