August LeetCoding Challenge Week 1: Day 2

my_hash_set.kt

class MyHashSet() {
    
    val numBuckets: Int = 97
    val buckets = List<MutableSet<Int>>(numBuckets){ mutableSetOf<Int>() }
    val hashFunc : (Int) -> Int = { key: Int -> key % numBuckets }
    
    fun add(key: Int) {
        buckets.get(hashFunc(key)).add(key)
    }

    fun remove(key: Int) {
        buckets.get(hashFunc(key)).remove(key)
    }

    /** Returns true if this set contains the specified element */
    fun contains(key: Int): Boolean {
        return buckets.get(hashFunc(key)).contains(key)
    }

}