Haskell generate subsets

orion picture orion · Nov 4, 2013 · Viewed 7.6k times · Source

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]]

Answer

Daniel Gratzer picture Daniel Gratzer · Nov 4, 2013

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