Étant donné une liste d'entrée (1,2,3,4,5), écrire une fonction qui peut être transmise à flatMap, et retourner le résultat Liste (1,2,2,3,3,4,4,5)

Question

Pour une entrée List(1,2,3,4,5), écrivez une fonction qui peut être transmise à flatMap, et renvoyez le résultat List(1,2,2,3,3,4,4,5).

La solution que j'ai écrit est la suivante:

object Main extends App { 
    val intList = List(1,2,3,4,5) 
    def returnTwo(arg: Int): List[Int] = List(arg, arg + 1) 
    val newList = intList.flatMap(returnTwo).filter(_ <= 5) 
    println(newList) 
} 

sortie:

List(1, 2, 2, 3, 3, 4, 4, 5, 5) 

Comment puis-je faire face à la 5 supplémentaire dans la liste, tout en collant au problème initial?

Answer 1

Si la liste d'entrée est toujours la liste donnée, c'est-à-dire Liste (1,2,3,4,5), nous pouvons alors coder en dur un peu, ce que vous avez déjà fait avec votre filtre.

object Main extends App { 
    val intList = List(1,2,3,4,5) 
    def returnTwo(arg: Int): List[Int] = { 
     if(arg == 1 || arg == 5) List(arg) else List(arg,arg) 
    } 
    val newList = intList.flatMap(returnTwo) 
    println(newList) 
} 

Answer 2

Comme mentionné par @ sepp2k, votre exemple a 5 dans la liste initiale. Voici une autre implémentation qui ne nécessite pas de flatMap ou de filtrage.

def returnTwo (list: List[Int]): List[Int] = { 
    list match { 
    case Nil => List.empty[Int] 
    case head :: tail => List(head, head + 1) ++ returnTwo(tail) 
    } 
} 

returnTwo(List(1,2,3,4)) 

Answer 3

vous avez déjà implémenté la fonction. Votre seul problème est le supplément 5. Vous pouvez le traiter en utilisant la fonction init, qui prend tout l'élément sauf le dernier élément d'un List.

val newList = intList.flatMap(returnTwo).filter(_ <= 5).init 

Answer 4

intList.head +: intList.init.tail.flatMap(returnTwo) :+ intList.last