推广mapAndUnzip

Question

我正在寻找一种简单的方法来编写函数

mapAndUnzip :: (Functor f) => (a -> (b,c)) -> f a -> (f b, f c)

我并不完全相信Functor是一个足够强的约束，但我会将它用于具体化。当f有类型(举几个例子) []、Data.Vector.Unboxed.Vector以及我自己围绕[a]和Vector a的包装类型时，我希望能够应用这个函数。(其他可能的类型包括Array、Repa向量等)

我的主要要求是，我不应该需要像(Unbox (b,c))这样的约束，只需要(Unbox b,Unbox c)。子需求:对每个输入元素只计算一次函数。

在对输入进行映射时，我看到了一种为Vector构建两个可变向量的方法，但我希望有一种比为不同类型创建一个新类和我自己的实例更好的方法。GHC.Util定义mapAndUnzip的特定列表方式让我认为一个通用解决方案可能是不可能的，但我想在侵入自己的解决方案之前，我会得到第二种意见。

Answer 1

Functor就足够了，您可以：

(编辑:只计算g一次)

mapAndUnzip g fa = (fmap fst fbc, fmap snd fbc)
  where
    fbc = fmap g fa