LeetCode 2. Add Two Numbers 两数相加 Java实现

题目：

给定两个非空链表来表示两个非负整数。位数按照逆序方式存储，它们的每个节点只存储单个数字。将两数相加返回一个新的链表。

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

示例：

输入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出：7 -> 0 -> 8
原因：342 + 465 = 807

这道题目本质上是一个大数相加问题，主要的处理思路就是把两个链表的对应位相加，然后把进位存在变量里面用于下一位的相加，和平时我们手算加法的步骤差不多。主要注意两种情况：

两个链表的长度不一样
算完之后进位符不为0得再加一个节点

接下来就是上代码时间了

public class TwoNumAdd1 {
    class ListNode {
        int val;
        ListNode next;
        ListNode(int x) { val = x; }
    }
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode result = new ListNode(-1);
        ListNode curr = result;
        int flag = 0;
        while (l1 != null && l2 != null) {
            curr.next = new ListNode((l1.val + l2.val + flag) % 10);
            flag = (l1.val + l2.val + flag) / 10;
            curr = curr.next;
            l1 = l1.next;
            l2 = l2.next;
        }
        ListNode remaining = (l1 == null) ? l2 : l1;
        curr.next = remaining;
        while (flag != 0) {
            if (curr.next == null) {
                curr.next = new ListNode(flag);
                break;
            }
            int tmp = (curr.next.val + flag);
            curr.next.val = tmp % 10;
            flag = tmp / 10;
            curr = curr.next;
        }
        return result.next;
    }
}

这是我自己写的一个版本，思路和功能是没问题，但是看上去很冗余，冗余的代码（越长的代码）就越容易出错，事实上也是调试了很久==，所以还是要努力写优雅的解法。

参考了网上大神们的解法后优化的版本：

public class TwoNumAdd1 {
    class ListNode {
        int val;
        ListNode next;
        ListNode(int x) { val = x; }
    }
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        int carry = 0;
        ListNode result = new ListNode(-1);
        ListNode curr = result;
        while (l1 != null || l2 != null || carry != 0) {
            int sum = ((l1 == null) ? 0 : l1.val) + ((l2 == null) ? 0 : l2.val) + carry;
            curr.next = new ListNode(sum % 10);
            carry = sum / 10;
            curr = curr.next;
            l1 = (l1 == null) ? l1 : l1.next;
            l2 = (l2 == null) ? l2 : l2.next;
        }
        return result.next;
    }
}

代码量明显下降了，而且每步所做的目的也更加明确了。

LeetCode 2. Add Two Numbers 两数相加 Java实现

猜你喜欢