Haskell concat 在列表理解中复制

Question

我在以下练习中遇到了问题：

利用列表推导，定义一个具有以下签名的函数：

reproduce :: [Int] -> [Int]

此函数将列表中的每个数字与自身的 x 个副本交换。

例如：

input: reproduce[3,5,1]
output: [3,3,3,5,5,5,5,5,1]

我做了以下事情：

reproduce :: [Int] -> [Int]
reproduce xs = [ x | x <- [1,2..10 ] , x `elem` xs , replicate x x ]

我的思路是让 x 属于一个小区间，x 必须是原始列表的一个元素，因此只有在它确实属于原始列表时才会添加使用 elemx，并复制 x ， x次，得到自己的 x 个副本。

当我尝试加载此函数时，我收到以下错误消息：

Couldn't match expected type `Bool' with actual type `[Int]'
In the return type of a call of `replicate'
In the expression: replicate x x
In a stmt of a list comprehension: replicate x x

我不知道如何修复此错误，我假设它与 elem 部分有关，但我不知道是什么。

注意：这是一个初学者练习，因此应该以简单的方式解决。

Answer 1

列表理解由结果表达式 L 和一个或多个以逗号分隔的右手形式 R1、R2、... 组成，排列如下：

[ L | R1, R2, ... ]

每个右边的形式必须是（1）一个生成器：

pattern <- expression

或 (2) let 绑定：

let pattern = expression

或 (3) 过滤条件（返回 Bool 的表达式）：

expression

在您的代码中，我们有三种这样的形式。 x <- [1,2..10 ] 是一个生成器（案例 2），x `elem` xs 是一个过滤条件（案例 3），但 replicate x x 不符合这些条件。这就是错误消息的含义：Haskell 期望表达式的类型为 Bool 但 replicate 返回一个列表。

---

要真正解决问题，您可以先这样做：

notQuiteReproduce xs = [ replicate x x | x <- xs ]

现在notQuiteReproduce [3, 5, 1] 产生[[3,3,3],[5,5,5,5,5],[1]]。剩下的就是“展平”结果列表。这也可以通过列表理解来完成，但我不愿意直接给出解决方案。