function - 如何在map函数中连接列表-Haskell？

Question

我有两个功能:

fun1 :: Int -> [Int]
fun2 :: [Int] -> [Int]

fun2 接受 Int list 并使用 map 帮助将 fun1 应用于此列表的每个元素。但是fun1返回[Int]。所以，我有类型冲突。如何解决我的问题？

Answer 1

您可能需要结合使用 map 和 concat 来实现它。假设 fun1 和 fun2 是这样的:

fun1 :: Int -> [Int]
fun1 x = [x,x]

fun2 :: [Int] -> [Int]
fun2 = map (+ 1)

solution :: [Int] -> [Int]
solution xs = concat $ map fun1 (fun2 xs)

或者按照@CarstenKonig的建议，您可以使用concatMap

solution2 :: [Int] -> [Int]
solution2 xs = concatMap fun1 $ fun2 xs

可以进一步简化为:

solution2 :: [Int] -> [Int]
solution2 = concatMap fun1 . fun2