blob: 6666203259f80bb85ec1cfc98a687b1059cbe957 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
|
import Control.Arrow
combination :: Int -> [a] -> [([a],[a])]
combination 0 xs = [([],xs)]
combination _ [] = []
combination c (x:xs) = (map ((x:).fst &&& snd) (combination (c-1) xs)) ++ (map (fst &&& (x:).snd) (combination c xs))
group :: [Int] -> [a] -> [[[a]]]
group [n] xs = map ((:[]).fst) (combination n xs)
group (n:ns) xs = concatMap (\(comb,rest) -> map (comb:) (group ns rest)) (combination n xs)
group' [] = const [[]]
group' (n:ns) = concatMap (uncurry $ (. group' ns) . map . (:)) . combination n
|
