Comment faire pour zipper des listes seulement si elles sont de la même taille et échouer si leurs tailles diffèrent?

Question

Supposons que je suis en train d'écrire une fonction zip deux listes que si elles sont de la même taille et autrement, ne:

def foo(xs: List[Int], ys: List[Int]): Either[String, List[(Int, Int)]] = 
    if (xs.size == ys.size) Right(xs zip ys) else Left(s"$xs and $ys have different sizes") 

Il fonctionne, mais je n'aime pas utiliser if. Comment amélioreriez-vous le code ci-dessus?

Answer 1

Vous pouvez utiliser pattern matching si pour une raison quelconque, vous ne l'aimez pas if

def zip(as: List[Int], bs: List[Int]): Either[String, List[(Int, Int)]] = (as, bs) match { 
    case (Nil, Nil) => Right(Nil) 
    case (a :: as1, b :: bs1) => for { 
     t <- zip(as1, bs1) 
    } yield (a, b) :: t 
    case _ => Left(s"$as and $bs have different sizes") 
    } 

ou

def zip(as: List[Int], bs: List[Int]): Either[String, List[(Int, Int)]] = { 
    @tailrec 
    def loop(as1: List[Int], bs1: List[Int], acc: List[(Int, Int)]): Either[String, List[(Int, Int)]] = (as1, bs1) match { 
     case (Nil, Nil) => Right(acc) 
     case (a :: as2, b :: bs2) => loop(as2, bs2, (a, b) :: acc) 
     case _ => Left(s"$as and $bs have different sizes") 
    } 

    loop(as, bs, Nil).map(_.reverse) 
    }