Haskell:返回[1..n]中n个元素列表的列表

Question

我需要一个功能f :: (Integral n) => n -> [[n]].

返回的列表应包含n所有元素源自的所有长度列表[1..n].

例:

f 2 = [[1,1],[1,2],[2,1],[2,2]]

---

对于常数ns来说这是一个简单的问题:

f2 = [[a, b] | a <- [1..2], b <- [1..2]]
f3 = [[a, b, c] | a <- [1..3], b <- [1..3], c <- [1..3]]
f4 = [[a, b, c, d] | a <- [1..4], b <- [1..4], c <- [1..4], d <- [1..4]]

Answer 1

一个解决方案可以

f n = sequence . replicate n $ [1..n]

请注意,f 10将有10 ^ 10个元素

Answer 2

另一种方法可能是:

Prelude> import Control.Monad
Prelude Control.Monad> f n = replicateM n [1..n]
Prelude Control.Monad> f 2
[[1,1],[1,2],[2,1],[2,2]]