powerset function

Author: matthewleon

src/Data/Set.purs

@@ -24,6 +24,7 @@ module Data.Set
   , subset
   , properSubset
   , intersection
+  , powerset
   ) where
 
 import Prelude hiding (map)
@@ -176,3 +177,9 @@ intersection s1 s2 = fromFoldable $ runPure (runSTArray (emptySTArray >>= inters
         LT -> pure $ Loop {a: l + 1, b: r}
         GT -> pure $ Loop {a: l, b: r + 1}
       else pure $ Done acc
+
+-- | The set of all subsets of the given set
+powerset :: forall a. Ord a => Set a -> Set (Set a)
+powerset =
+  foldl (\subPowerset a -> subPowerset `union` map (insert a) subPowerset)
+        (singleton empty)

test/Test/Main.purs

@@ -4,10 +4,9 @@ import Prelude
 
 import Control.Monad.Eff (Eff)
 import Control.Monad.Eff.Console (CONSOLE, log)
-import Test.Assert (ASSERT, assert)
-
 import Data.Set (Set)
 import Data.Set as S
+import Test.Assert (ASSERT, assert)
 
 main :: Eff (console :: CONSOLE, assert :: ASSERT) Unit
 main = do
@@ -26,3 +25,16 @@ main = do
          s2 = S.fromFoldable [2,4,6,8,10]
          s3 = S.fromFoldable [2,4]
      assert $ S.intersection s1 s2 == s3
+
+  log "powerset"
+  do let set = S.fromFoldable [0, 1, 2]
+     assert $ S.powerset set == (S.fromFoldable `S.map` S.fromFoldable [
+                                    []
+                                  , [0]
+                                  , [1]
+                                  , [2]
+                                  , [0,1]
+                                  , [0,2]
+                                  , [1,2]
+                                  , [0,1,2]
+                                ])