如何在Rust中实现通用链表映射函数?

时间:2017-03-30 08:32:27

标签: generics polymorphism rust

我正在用Rust切割牙齿,我正在尝试实现一般类型的链表。到目前为止,我的conslen函数仍有效,但map出现了一些我无法弄清楚的错误。

use std::fmt;

#[derive(Debug)]
enum List<A> {
    Empty,
    Cons(A, Box<List<A>>),
}

fn cons<A>(x: A, xs: List<A>) -> List<A> {
    return List::Cons(x, Box::new(xs));
}

fn len<A>(xs: List<A>) -> i32 {
    match xs {
        List::Empty => 0,
        List::Cons(_, xs) => 1 + len(*xs),
    }
}

fn map<A, B>(f: &Fn(A) -> B, xs: List<A>) -> List<B> {
    match xs {
        List::Empty => List::Empty,
        List::Cons(x, xs) => cons(f(x), map(f, *xs)),
    }
}

fn main() {
    let xs = cons(1, cons(2, cons(3, List::Empty)));
    println!("{:?}", xs);
    println!("{:?}", len(xs));
    let f = |x: i32| x * x;
    println!("{:?})", map(f, xs));
}

错误

error[E0308]: mismatched types
  --> src/main.rs:32:27
   |
32 |     println!("{:?})", map(f, xs));
   |                           ^ expected reference, found closure
   |
   = note: expected type `&std::ops::Fn(_) -> _`
              found type `[closure@src/main.rs:31:13: 31:27]`

预期输出

Cons(1, Cons(2, Cons(3, Empty)))
3
Cons(1, Cons(4, Cons(9, Empty)))

我的特殊问题在于

println!("{:?})", map(f, xs));

如果我评论该行,前两行输出是正确的。我不确定我的map电话

有什么问题

更新

aochagavia帮助我理解了函数引用问题和第一个所有权问题(很明显!) - 我在使用len map中使用的相同技术时遇到了麻烦新错误

我更新的map功能如下所示

fn map<A, B>(f: &Fn(A) -> B, xs: &List<A>) -> List<B> {
    match *xs {
        List::Empty => List::Empty,
        List::Cons(x, ref xs) => cons(f(x), map(f, xs)),
    }
}

我现在正在尝试这个

let f = |x: i32| x * x;
let ys = map(&f, &xs);
let zs = map(&f, &xs);
println!("{:?})", ys);
println!("{:?})", zs);

新错误是

error[E0009]: cannot bind by-move and by-ref in the same pattern
  --> src/main.rs:23:20
   |
23 |         List::Cons(x, ref xs) => cons(f(x), map(f, xs)),
   |                    ^  ------ both by-ref and by-move used
   |                    |
   |                    by-move pattern here

1 个答案:

答案 0 :(得分:4)

错误消息很大,因为它发生在宏中,但是如果你添加它:let y = map(f, xs);你得到一个更短(并且稍微更准确)的一个:

error[E0308]: mismatched types
  --> <anon>:32:15
   |
32 |   let y = map(f, xs);
   |               ^ expected reference, found closure
   |
   = note: expected type `&std::ops::Fn(_) -> _`
              found type `[closure@<anon>:31:11: 31:25]`

也就是说,您通过值而不是引用传递闭包!使用map(&f, xs)(注意&符号)应该可以解决错误。但是,所有权还存在另一个问题(见下文)。

所有权问题

len函数的类型签名是fn len<A> (xs: List<A>) -> i32。这意味着它将获得列表的所有权以计算其长度。然而,这不是你想要的,因为它会阻止你以后使用这个列表!因此,您从编译器获得错误。

解决此问题的明智方法是让len借用xs而不是消费它。像这样:

fn len<A>(xs: &List<A>) -> i32 {
    match *xs {
        List::Empty => 0,
        List::Cons(_, ref xs) => 1 + len(xs),
    }
}

最后,您需要修改main功能以反映此更改,方法是调用lenlen(&xs)(请注意&符号,您可以将其视为借用运算符)。

使地图借用xs

正如naomik在评论中指出的那样,map似乎也是借用xs而不是消费它的候选人。可能的实现方式是:

fn map<A, B>(f: &Fn(&A) -> B, xs: &List<A>) -> List<B> {
    match *xs {
        List::Empty => List::Empty,
        List::Cons(ref x, ref xs) => cons(f(x), map(f, xs)),
    }
}

与原始版本的主要区别在于,闭包现在需要&A而不是A(请参阅Fn(&A) -> B)。这很自然,因为不可能消耗借入中包含的值(这意味着借用机制完全被破坏)。

在主要内容中,您需要像这样调用map

let f = |x: &i32| (*x) * (*x);
map(&f, &xs);

请注意f现在借用其参数而不是使用它,正如map的类型签名所要求的那样。

闭包中的一些额外背景

在Rust中,闭包有点特殊。您可以使用一个很好的语法构造它们,但最后它们只是碰巧实现FnFnMutFnOnce特征的结构。

如果您希望按值传递它们(而不是像在代码中那样通过引用传递),可以使用以下类型签名使map函数通用:

fn map<F, A, B> (f: F, xs: List<A>) -> List<B>
    where F: Fn(A) -> B
{

这也为您提供静态调度。如果您想了解更多信息,请阅读trait objects and static/dynamic dispatch