Skip to content

Instantly share code, notes, and snippets.

@huydx
Created April 7, 2013 07:35
Show Gist options
  • Save huydx/5329449 to your computer and use it in GitHub Desktop.
Save huydx/5329449 to your computer and use it in GitHub Desktop.
scala coursera course week 2
package funsets
import common._
/**
* 2. Purely Functional Sets.
*/
object FunSets {
/**
* We represent a set by its characteristic function, i.e.
* its `contains` predicate.
*/
type Set = Int => Boolean
/**
* Indicates whether a set contains a given element.
*/
def contains(s: Set, elem: Int): Boolean = s(elem)
/**
* Returns the set of the one given element.
*/
def singletonSet(elem: Int): Set =
return (x: Int) =>
x == elem
/**
* Returns the union of the two given sets,
* the sets of all elements that are in either `s` or `t`.
*/
def union(s: Set, t: Set): Set =
return (x: Int) =>
(contains(s, x) || contains(t, x))
/**
* Returns the intersection of the two given sets,
* the set of all elements that are both in `s` and `t`.
*/
def intersect(s: Set, t: Set): Set =
return (x: Int) =>
(contains(s, x) && contains(t, x))
/**
* Returns the difference of the two given sets,
* the set of all elements of `s` that are not in `t`.
*/
def diff(s: Set, t: Set): Set =
return (x: Int) =>
(contains(s, x) && !(contains(t, x)))
/**
* Returns the subset of `s` for which `p` holds.
*/
def filter(s: Set, p: Int => Boolean): Set =
return (x: Int) => p(x)
/**
* The bounds for `forall` and `exists` are +/- 1000.
*/
val bound = 1000
/**
* Returns whether all bounded integers within `s` satisfy `p`.
*/
def forall(s: Set, p: Int => Boolean): Boolean = {
def iter(a: Int): Boolean = {
if (a > bound) true
else if (contains(s, a) && !p(a)) false
else iter(a+1)
}
iter(-bound)
}
/**
* Returns whether there exists a bounded integer within `s`
* that satisfies `p`.
*/
def exists(s: Set, p: Int => Boolean): Boolean = !forall(s, x => !p(x))
/**
* Returns a set transformed by applying `f` to each element of `s`.
*/
def map(s: Set, f: Int => Int): Set = {
def iterate(s:Set, accumulateSet: Set, a:Int): Set = {
if (a > bound) accumulateSet
else if (contains(s, a)) iterate(s, union(accumulateSet, x=>f(a) == x), a+1)
else iterate(s, accumulateSet, a+1)
}
iterate(s, x=>false, -3)
}
/**
* Displays the contents of a set
*/
def toString(s: Set): String = {
val xs = for (i <- -bound to bound if contains(s, i)) yield i
xs.mkString("{", ",", "}")
}
/**
* Prints the contents of a set on the console.
*/
def printSet(s: Set) {
println(toString(s))
}
}
@mattlevan
Copy link

There is a way to implement map with just one line!

@itamarjo
Copy link

itamarjo commented Mar 26, 2017

There is a way to implement map with just one line!

can you share that line? :-) @mattlevan

@wayoung
Copy link

wayoung commented Apr 4, 2017

def map(s: Set, f: Int => Int): Set = (i: Int) => exists(s, (j: Int) => i == f(j))

@valtih1978
Copy link

Why -3 in the map?

@dsu4rez
Copy link

dsu4rez commented Feb 20, 2018

I think the correct value is -bound instead of -3 @valtih1978

@ryszard314159
Copy link

How about:
def map(s: Set, f: Int => Int): Set = (x: Int) => contains(map(s, f), f(x))

@mrseanryan
Copy link

def map(s: Set, f: Int => Int): Set = (x: Int) => contains(map(s, f), f(x))

This failed my tests ...

is that an infinite loop? 🤔

@AntoniaMouawad
Copy link

def map(s: FunSet, f: Int => Int): FunSet = y => exists(s, x => y == f(x))
translates to when you get a y, check if there is a value x in my set, for which f(x)== y

@koolkrik
Copy link

koolkrik commented Aug 4, 2020

there is a bug in this code
filter should be defined as :
def filter(s: FunSet, p: Int => Boolean): FunSet = (a: Int) => if (contains(s,a)) p(a) else false

otherwise it returns all elements within the bounds. the contains check is necessary, we cannot just blindly apply p(a)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment