面试题02.05. 链表求和

面试题 02.05. 链表求和

难度：中等

给定两个用链表表示的整数，每个节点包含一个数位。

这些数位是反向存放的，也就是个位排在链表首部。

编写函数对这两个整数求和，并用链表形式返回结果。

示例：

输入：(7 -> 1 -> 6) + (5 -> 9 -> 2)，即617 + 295
输出：2 -> 1 -> 9，即912

**进阶：**思考一下，假设这些数位是正向存放的，又该如何解决呢?

示例：

输入：(6 -> 1 -> 7) + (2 -> 9 -> 5)，即617 + 295
输出：9 -> 1 -> 2，即912

解答：

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    //模拟加法运算
    //时间复杂度O(max(m,n)),空间复杂度O(1)。
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode head = null ,tail = null;
        int carry = 0;
        while(l1 != null || l2 != null){
            int a = l1 == null ? 0 : l1.val;
            int b = l2 == null ? 0 : l2.val;
            int sum = a + b + carry;
            if(head == null){
                head = tail = new ListNode(sum % 10);
            }else{
                tail.next = new ListNode(sum % 10);
                tail = tail.next;
            }
            carry = sum / 10;
            if(l1 != null) l1 = l1.next;
            if(l2 != null) l2 = l2.next;
        }
        tail.next = carry > 0 ? new ListNode(1) : null;
        return head;
    }
}

class Solution {
    //递归
    //时间复杂度O(N)。空间复杂度O(1)。
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        return addTwoNumbers(l1, l2, 0);
    }

    public ListNode addTwoNumbers(ListNode l1, ListNode l2, int carry){
        if(l1 == null && l2 == null && carry == 0) return null;
        int a = l1 == null ? 0 : l1.val;
        int b = l2 == null ? 0 : l2.val;
        int sum = a + b + carry;
        carry = sum / 10;
        ListNode node = new ListNode(sum % 10);
        if(l1 == null) l1 = null;
        else l1 = l1.next;
        if(l2 == null) l2 = null;
        else l2 = l2.next;
        node.next = addTwoNumbers(l1, l2, carry);
        return node;
    }
}

参考自：

作者：LeetCode-Solution
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。