Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Haskell generate subsets

I have a function 'subsets' which generate all the subsets of a given set:

subsets :: [Int] -> [[Int]]
subsets []  = [[]]
subsets (x:xs) = subsets xs ++ map (x:) (subsets xs)

How can I combine map, foldl and filter in another function to return me all the subsets with elements that sum up to 0?

**Example: **

set    = [1,-1,5,2,-2,3]
result = [[1,-1],[2,-2],[-1,-2,3]]
like image 635
orion Avatar asked Nov 04 '13 16:11

orion


1 Answers

You have subsets already. So we need a function

filterSubs :: [[Int]] -> [[Int]]
filterSubs = --remove all subsets which don't sum to 0

So next we'd need a predicate

sumZero :: [Int] -> Bool
sumZero xs = sum xs == 0

Now, using this and filter it's easy to construct filterSubs. I'll leave this to you to figure out exactly how that works. And then our solution is trivial

zeroSubs = filterSubs . subsets
like image 52
Daniel Gratzer Avatar answered Sep 29 '22 19:09

Daniel Gratzer