LeetCode 002 Add Two Numbers

学习记录

问题

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

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

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

解答

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode retList = new ListNode(0);
        ListNode p = l1, q = l2, r = retList;
        boolean flag = false;
        while(p != null || q != null) {
            int x = (p != null) ? p.val : 0;
            int y = (q != null) ? q.val : 0;
            int temp = x + y;
            if(flag) {
                temp += 1;
                flag = false;
            }
            if(temp > 9) {
                flag = true;
                temp -= 10;
            }
            r.next = new ListNode(temp);
            r = r.next;
            if(p != null) p = p.next;
            if(q != null) q = q.next;
        }
        if(flag) {
            r.next = new ListNode(1);
        }
        return retList.next;
    }
}

复杂度分析：

• 时间复杂度：$O(max(m, n))$
• 空间复杂度：$O(max(m, n))$，新链表的长度最多为$max(m, n) + 1$。