2013-05-29 125 views
8

我偶爾會打這樣的代碼:如何分割F [A /B]進入(F [A],F [B])

val things : List[A \/ B] = ??? 
val (as, bs) : (List[A], List[B]) = ??? //insert something to do this 

或在我目前的情況下,我想Map[A, B \/ C] => (Map[A, B], Map[A, C])

有一個很好的方法可以在一般情況下做到這一點F[A \/ B]與適當的限制F?它看起來隱約像是Unzip主題的變體。

回答

9

下面是我們如何處理這個問題,也可以是Either和Validation,而不僅僅是列表,但是其他的Foldable。

object Uncozip { 
    implicit val wtf = language.higherKinds 

    // Typeclass which covers sum types such as \/, Either, Validation 
    trait Sum2[F[_, _]] { 
    def cata[A, B, X](a: A ⇒ X, b: B ⇒ X)(fab: F[A, B]): X 
    } 

    implicit val sumEither: Sum2[Either] = new Sum2[Either] { 
    def cata[A, B, X](a: A ⇒ X, b: B ⇒ X)(fab: Either[A, B]): X = { 
     fab match { 
     case Left(l) ⇒ a(l) 
     case Right(r) ⇒ b(r) 
     } 
    } 
    } 

    implicit val sumEitherz: Sum2[\/] = new Sum2[\/] { 
    def cata[A, B, X](a: A ⇒ X, b: B ⇒ X)(fab: A \/ B): X = { 
     fab.fold(a(_), b(_)) 
    } 
    } 

    implicit val sumValidation: Sum2[Validation] = new Sum2[Validation] { 
    def cata[A, B, X](a: A ⇒ X, b: B ⇒ X)(fab: A Validation B): X = { 
     fab.fold(a(_), b(_)) 
    } 
    } 

    abstract class Uncozips[F[_], G[_, _], A, B](fab: F[G[A, B]]) { 
    def uncozip: (F[A], F[B]) 
    } 

    implicit def uncozip[F[_]: Foldable, G[_, _], A, B](fab: F[G[A, B]])(implicit g: Sum2[G], mfa: ApplicativePlus[F], mfb: ApplicativePlus[F]): Uncozips[F, G, A, B] = new Uncozips[F, G, A, B](fab) { 
    def uncozip = { 
     implicitly[Foldable[F]].foldRight[G[A, B], (F[A], F[B])](fab, (mfa.empty, mfb.empty)) { (l, r) ⇒ 
     g.cata[A, B, (F[A], F[B])]({ (a: A) ⇒ (mfa.plus(mfa.point(a), r._1), r._2) }, 
      { (b: B) ⇒ (r._1, mfa.plus(mfa.point(b), r._2)) })(l) 
     } 
    } 
    } 
} 
+0

由於我們最終將Map看作一個Iterable [(A,B \/C)],然後將它分解爲A和B \/C,但它仍然非常有用,因此不便於覆蓋地圖的情況。 – AlecZorab

2

您還可以在地圖things(Option[A], Option[B])列表,解壓縮在兩個列出清單,然後團結所產生的名單:

import scalaz._ 
import Scalaz._ 

val things: List[String \/ Int] = List("foo".left, 42.right) 
val (strs, ints): (List[String], List[Int]) = things. 
    map { d => (d.swap.toOption, d.toOption) }. // List[(Option[String], Option[Int])] 
    unzip.          // (List[Option[String]], List[Option[Int]]) 
    bimap(_.unite, _.unite)      // (List[String], List[Int]) 

這不是特別有效,由於遍歷列表三倍。

1

這裏是一個辦法(用於列表):

val things : List[A \/ B] = ??? 
val (as, bs) = (things.map(_.swap.toList).join, things.map(_.toList).join) 

而對於地圖:

val things: Map[String, String \/ Int] = ??? 
val (as, bs) = (things.mapValues(_.swap.toList).filterNot(e => e._2.isEmpty), 
       things.mapValues(_.toList).filterNot(e => e._2.isEmpty)) 

我有一個很難拿出一個辦法來概括這個通過任何F(我相信你需要MonoidApplicative的實例爲F)。

相關問題