对两个链表求和

问题描述

我有一个这样的节点:

struct ListNode {
    int val;
    ListNode *next;
    ListNode() 
       : val(0),next(nullptr) {}
    ListNode(int x) 
       : val(x),next(nullptr) {}
    ListNode(int x,ListNode *next) 
       : val(x),next(next) {}
};

我想使用递归对两个链接列表求和。我不确定,但我猜是尾递归。

输入: l1 = [2,4,3],l2 = [5,6,4]

输出 [7,8]

解释: 342 + 465 = 807。

这是我的代码

ListNode* addTwoNumbers(ListNode* l1,ListNode* l2,int carry = 0,ListNode *result = nullptr) {
    int n1 = l1 != nullptr ? l1->val : 0;
    int n2 = l2 != nullptr ? l2->val : 0;
    int val = (n1 + n2 + carry) % 10;
    if(result == nullptr){
        result = new ListNode(val);
    }else{
        result->next = new ListNode(val);
        result = result->next;
    }
    result = addTwoNumbers(l1->next != nullptr ? l1->next : nullptr,l2->next != nullptr ? l2->next : nullptr,(n1 + n2 + carry) - 10,result);
    return result;
}

但是我遇到了分段错误,即使在传递两个单节点列表时也是如此:

ListNode * a = new ListNode(1);
ListNode * b = new ListNode(2);
ListNode * c = addTwoNumbers(a,b); 
std::cout << c->val << "\n";

我看不出有什么问题,它应该可以工作。我错过了什么?

解决方法

有几个问题:

  • 你的递归永远不会停止,这就是分段错误的原因——在递归调用中你评估 l1->next,但在某个点 l1 将是一个空指针,所以你得到例外。缺少递归的所谓“基本情况”。当两个列表指针都为空且没有进位时,不应该递归调用,而是应该返回一个空指针
  • (n1 + n2 + carry) - 10 是错误的公式。应该是 (n1 + n2 + carry) / 10
  • result = result->next; 是错误的,因为您将永远失去对刚刚创建的节点的引用。创建新列表的头部后,该头部不需要再次更改。当您首先将参数的头部值加在一起时,该总和也应该首先出现在结果中,并且对第一个节点的引用是函数应该返回的。

还有:

  • 应该没有必要将 result 作为参数传递。确保递归调用为您作为参数传递的较短列表返回一致的结果(链接列表)。然后,您“唯一”要做的就是将当前总和作为一个节点添加 到返回的列表中,然后您就完成了。所以实际上不需要 result 参数。

这是更正后的代码:

ListNode* addTwoNumbers(ListNode* l1,ListNode* l2,int carry = 0) {
    // base case
    if (l1 == nullptr && l2 == nullptr && carry == 0) return nullptr;
    int n1 = l1 != nullptr ? l1->val : 0;
    int n2 = l2 != nullptr ? l2->val : 0;
    int sum = n1 + n2 + carry;
    return new ListNode(sum % 10,addTwoNumbers(l1 != nullptr ? l1->next : nullptr,l2 != nullptr ? l2->next : nullptr,sum / 10));
}