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的功能應該是這樣的 謝謝！

它與這個問題有關： Haskell 中的 Verbal Arithmetics (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]]

Answer 2

威爾replicateM你需要什么？

Prelude Control.Monad> take 10 $ replicateM 3 [0..9]
[[0,0,0],[0,0,1],[0,0,2],[0,0,3],[0,0,4],[0,0,5],[0,0,6],[0,0,7],[0,0,8],[0,0,9]]

Prelude Control.Monad> take 10 $ replicateM 4 [1,3,3,7]
[[1,1,1,1],[1,1,1,3],[1,1,1,3],[1,1,1,7],[1,1,3,1],[1,1,3,3],[1,1,3,3],[1,1,3,7],[1,1,3,1],[1,1,3,3]]

Prelude Control.Monad> take 10 $ replicateM 2 [4,2]
[[4,4],[4,2],[2,4],[2,2]]