有没有办法递归地压扁元组？

Posted 2021-04-09

在Rust中，有没有办法使用traits和impls（递归地）展平元组？

如果它有帮助，那么适用于N个嵌套对的东西是一个好的开始

trait FlattenTuple {
    fn into_flattened(self) -> /* ??? */
}

// such that
assert_eq!((1, (2, 3)).into_flattened(), (1, 2, 3))

如果可以使用任何类型的嵌套元组进行扩展工作，那将会更好：

assert_eq!(((1, 2), 2, (3, (4, 5))).into_flattened(), (1, 2, 2, 3, 4, 5))

答案

也许对于某些“扁平化”的小定义，但实际上并非如此。

从最具体的实现开始：

trait FlattenTuple {
    fn into_flattened(self) -> (u8, u8, u8);
}

impl FlattenTuple for (u8, (u8, u8)) {
    fn into_flattened(self) -> (u8, u8, u8) {
        (self.0, (self.1).0, (self.1).1)
    }
}

然后使它更通用：

trait FlattenTuple {
    type Output;
    fn into_flattened(self) -> Self::Output;
}

impl<A, B, C> FlattenTuple for (A, (B, C)) {
    type Output = (A, B, C);

    fn into_flattened(self) -> Self::Output {
        (self.0, (self.1).0, (self.1).1)
    }
}

然后重复每个可能的排列：

impl<A, B, C, D, E, F> FlattenTuple for ((A, B), C, (D, (E, F))) {
    type Output = (A, B, C, D, E, F);

    fn into_flattened(self) -> Self::Output {
        ((self.0).0, (self.0).1, self.1, (self.2).0, ((self.2).1).0, ((self.2).1).1)
    }
}

这两个实现涵盖了您的两种情况。

但是，您必须枚举您想要的每种输入类型，可能是通过代码生成。我无法“检查”输入类型，然后将其“拼接”到输出类型中。

你甚至可以尝试写一些有点递归的东西：

impl<A, B, C, D, E, F> FlattenTuple for (A, B)
    where A: FlattenTuple<Output = (C, D)>,
          B: FlattenTuple<Output = (E, F)>,
{
    type Output = (C, D, E, F);

    fn into_flattened(self) -> Self::Output {
        let (a, b) = self;
        let (c, d) = a.into_flattened();
        let (e, f) = b.into_flattened();

        (c, d, e, f)
    }
}

但这很快就会遇到基本问题：终端42没有实现FlattenTuple，如果你尝试impl<T> FlattenTuple for T，你会遇到冲突的特征实现。

另一答案

我用specialized和auto trait实现了这个。

docs.rs

github repo

---

用法基本上就是

assert_eq!((1, (2, ((3,) (4, 5)), (), 6).flatten(), (1, 2, 3, 4, 5, 6));

它删除所有空单位元组()（如python）。

---

如果您正在编写通用代码，则需要添加

where (A, B): Flatten

rustc想要这个，因为只有当输出元组的长度小于13时才实现Flatten。