2. 两数相加

257 阅读2分钟

题目介绍

力扣第2题:leetcode-cn.com/problems/ad…

image.png

image.png

分析

由于输入的两个链表都是逆序存储数字的位数的,因此两个链表中同一位置的数字可以直接相加。

我们同时遍历两个链表,逐位计算它们的和,并与当前位置的进位值相加。具体而言,如果当前两个链表处相应位置的数字为n1 , n2,进位值为 carry,则它们的和为 n1 + n2 + carry;其中,答案链表处相应位置的数字为 (n1 + n2 + carry) mod 10,而新的进位值为 (n1 + n2 + carry) / 10

如果两个链表的长度不同,则可以认为长度短的链表的后面有若干个0 。

此外,如果链表遍历结束后,有carry > 0,还需要在答案链表的后面附加一个节点,节点的值为carry。

代码如下:

/**
 * 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 pre = new ListNode(0);
        ListNode cur = pre;
        //存储进位
        int carry = 0;
        while(l1 != null || l2 != null) {
            int x = l1 == null ? 0 : l1.val;
            int y = l2 == null ? 0 : l2.val;
            int sum = x + y + carry;
            
            //更新进位的值
            carry = sum / 10;
            //获取出当前位数相加之后去掉进位的值
            sum = sum % 10;
            cur.next = new ListNode(sum);

            cur = cur.next;
            
            if(l1 != null) {
                l1 = l1.next;
            }
            
            if(l2 != null) {
                l2 = l2.next;
            }
        }
        //如果最后进位还有值,说明相加之后,位数增加了1,还需要补一个最后节点
        if(carry == 1) {
            cur.next = new ListNode(carry);
        }
        return pre.next;
    }
}

复杂度分析

  • 时间复杂度:O(max(m,n)),其中 m 和 n 分别为两个链表的长度。我们要遍历两个链表的全部位置,而处理每个位置只需要 O(1)的时间。

  • 空间复杂度:O(1)。注意返回值不计入空间复杂度。