具有输出列表长度的 Haskell 排列

Question

我有这样的代码来创建排列:

--unique permutation
perm :: [t] -> [[t]]
perm [] = [[]]
perm (x:xs) = [(y:zs) | (y,ys) <- views (x:xs), zs <- perm ys]

--create all possible views
views :: [t] -> [(t,[t])]
views [] = []
views (x:xs) = ((x,xs) : [ (y,(x:ys)) | (y,ys) <- views xs ])

但是我想将输出限制在一定长度内。例如，它应该采用一个参数来指定排列输出中元素的数量。我想创建从列表 [0..9] 中提取的 k 个不同的排列。 因此，例如，如果 k 等于 3，则输出应如下所示:[1,2,3]、[9,8,7]、[0,6,8] 等。

现在，如果我将 [0..9] 传递给函数，它将生成一个排列长度为 10 的列表。我正在努力想出一个解决方案。 最后，该函数应该类似于 perm k list 谢谢!

它与这个问题有关:Verbal Arithmetics in Haskell (SEND + MORE = MONEY)

Answer 1

你的意思是这样的吗？

import Data.List (permutations)

choose n list = concatMap permutations $ choose' list [] where
  choose' []     r = if length r == n then [r] else []
  choose' (x:xs) r | length r == n = [r]
                   | otherwise     = choose' xs (x:r) 
                                  ++ choose' xs r

输出:

*Main> choose 2 [0..5]
[[1,0],[0,1],[2,0],[0,2],[3,0],[0,3],[4,0],[0,4],[5,0],[0,5],[2,1]
,[1,2],[3,1],[1,3],[4,1],[1,4],[5,1],[1,5],[3,2],[2,3],[4,2],[2,4]
,[5,2],[2,5],[4,3],[3,4],[5,3],[3,5],[5,4],[4,5]]