难度:中等
题目:
2. 两数相加
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例 1:
输入:l1 = [2,4,3], l2 = [5,6,4]
输出:[7,0,8]
解释:342 + 465 = 807.
示例 2:
输入:l1 = [0], l2 = [0]
输出:[0]
示例 3:
输入:l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
输出:[8,9,9,9,0,0,0,1]
思路:
初始化一个结果链表的头节点 resultHead 和一个当前节点 resultCur,初始值均为 null。
- 初始化一个进位变量 carry,初始值为 0,用于存储两个数字相加的进位值。
- 遍历两个链表,每次取出对应节点的值相加,并加上进位值。
- 如果有一个链表已经遍历完,则将另一个链表剩余的部分直接加到结果链表中。
- 否则,将两个节点的值相加,并加上进位值。
- 将相加的结果除以 10,得到进位值和当前位的值。
- 创建一个新节点,将当前位的值赋给新节点的值,并将新节点连接到结果链表的尾部。
- 更新进位值为计算得到的进位值。
- 将当前节点指向新节点。
- 继续遍历下一个节点,重复步骤 3-7,直到两个链表都遍历完。
- 如果最后的进位值不为 0,创建一个新节点,将进位值赋给新节点的值,并将新节点连接到结果链表的尾部。
- 返回结果链表的头节点 resultHead。
class Solution { public ListNode addTwoNumbers(ListNode l1, ListNode l2) { ListNode resultHead = null; ListNode resultCur = null; int carry = 0; while (l1 != null || l2 != null) { int sum = carry; if (l1 != null) { sum += l1.val; l1 = l1.next; } if (l2 != null) { sum += l2.val; l2 = l2.next; } carry = sum / 10; ListNode newNode = new ListNode(sum % 10); if (resultHead == null) { resultHead = newNode; resultCur = newNode; } else { resultCur.next = newNode; resultCur = resultCur.next; } } if (carry != 0) { resultCur.next = new ListNode(carry); } return resultHead; } }
时间复杂度:O(max(N, M))
空间复杂度:O(max(N, M))