You are given two linked lists representing two non-negative numbers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.
Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8
Subscribe to see which companies asked this question
拿到这个题目很自然地就想到利用一个布尔变量carry来判断逐个加的过程中是否有进位产生。
难点在于,当其中一个链表已经没有后续的时候,该如何继续操作。
刚开始的想法是直接把另一个链表剩下的数据链接到结果上,再判断之前是否有进位而决定是否在next->val上加1。
然而提交后的WA却让我发现,当另一个链表剩下一堆 9 -> 9 -> 9的时候,这不是个好的思路。
于是想到再添加一个新的递归函数 justCarry 来处理进位产生的问题,然而没等写完就感觉这样的打补丁式的修补WA带来的问题很弱智。
于是重新思考,添加了当 当前 输入为NULL时生成一个0结点的想法。于是有了以下AC 36ms Solution:
class Solution {
public:
ListNode* add(ListNode* l1, ListNode* l2, bool carryBit)
{
if (l1 != NULL || l2 != NULL)
{
if (l1 == NULL)
{
l1 = new ListNode (0);
}
if (l2 == NULL)
{
l2 = new ListNode (0);
}
l1->val += l2->val;
if (carryBit)
{
l1->val += 1;
}
carryBit = l1->val / 10;
l1->val %= 10;
l1->next = add(l1->next, l2->next, carryBit);
}
else if (carryBit)
{
ListNode* newBit = new ListNode (1);
return newBit;
}
return l1;
}
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2)
{
l1 = add(l1, l2, false);
return l1;
}
};
其实最开始用的是 if (l1 -> val >= 10) carry = true;
后来发现这种解法,
现在看起来,直接把carry定义为一个int或许更好,这样的话,连l1 - > val += 1都可以合并到上面去,少一个判断结构。
之后从LeetCode官网找到了非递归实习版本。如下(Java):
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
ListNode dummyHead = new ListNode(0);
ListNode p = l1, q = l2, curr = dummyHead;
int carry = 0;
while (p != null || q != null) {
int x = (p != null) ? p.val : 0;
int y = (q != null) ? q.val : 0;
int sum = carry + x + y;
carry = sum / 10;
curr.next = new ListNode(sum % 10);
curr = curr.next;
if (p != null) p = p.next;
if (q != null) q = q.next;
}
if (carry > 0) {
curr.next = new ListNode(carry);
}
return dummyHead.next;
}