rust - ListNode 的下一个字段类型为 Option>，对吗？

Question

我试图解决https://leetcode.com/problems/add-two-numbers，这个问题很容易，但由于借用问题我无法完成它。我尝试了几个 3 小时，但我怀疑ListNode的next字段类型Option<Box<ListNode>>不正确。

当我切换到 c# 时，我很快就解决了这个问题。以下是c#版本的解决方案。我无法将其翻译为生锈。

public class Solution {
    public ListNode AddTwoNumbers(ListNode l1, ListNode l2) {
        ListNode re = null;
        ListNode next = null;
        int carry = 0;
        while(l1 != null || l2 != null) {
            var val = (l1 != null ? l1.val : 0) + (l2 != null ? l2.val : 0) + carry;
            carry = val / 10;
            val %= 10;
            if (re != null){
                next.next = new ListNode(val);
                next = next.next;
            } else {
                re = new ListNode(val);
                next = re;
            }
            if (l1 != null) {
                l1 = l1.next;
            }
            if (l2 != null) {
                l2 = l2.next;
            }
        }
        if (carry > 0) {
            next.next = new ListNode(carry);
        }
        return re;
    }
}

标签： rust

Answer 1

根据@rodrigo 的建议。我终于以递归函数的方式解决了这个问题。

fn internal_add_two_number(
    mut l1: &Option<Box<ListNode>>,
    mut l2: &Option<Box<ListNode>>,
    mut carry: i32,
) -> Option<Box<ListNode>> {
    let mut sum = carry;
    if l1.is_none() && l2.is_none() {
        if carry > 0 {
            return Some(Box::new(ListNode::new(carry)));
        }
        return None;
    }
    if let Some(p) = l1 {
        sum += p.val;
        l1 = &p.next;
    }
    if let Some(p) = l2 {
        sum += p.val;
        l2 = &p.next;
    }
    carry = sum / 10;
    sum = sum % 10;

    Some(Box::new(ListNode {
        val: sum,
        next: internal_add_two_number(l1, l2, carry),
    }))
}

impl Solution {
    pub fn add_two_numbers(
        l1: Option<Box<ListNode>>,
        l2: Option<Box<ListNode>>,
    ) -> Option<Box<ListNode>> {
        internal_add_two_number(&l1, &l2, 0)
    }
}