|
| 1 | +/** |
| 2 | + * Question Link: https://leetcode.com/problems/design-hashmap/ |
| 3 | + * Primary idea: Use modulo and array of list / linked list to handle handle function and collision. |
| 4 | + * Time Complexity: O(n), Space Complexity: O(n) |
| 5 | + */ |
| 6 | + |
| 7 | +class MyHashMap { |
| 8 | + |
| 9 | + let keySpace = 2069 |
| 10 | + var buckets: [[(Int, Int)]] |
| 11 | + |
| 12 | + /** Initialize your data structure here. */ |
| 13 | + init() { |
| 14 | + buckets = Array(repeating: [(Int, Int)](), count: keySpace) |
| 15 | + } |
| 16 | + |
| 17 | + /** value will always be non-negative. */ |
| 18 | + func put(_ key: Int, _ value: Int) { |
| 19 | + var bucket = buckets[key % keySpace] |
| 20 | + |
| 21 | + if let index = find(key, bucket) { |
| 22 | + bucket[index].1 = value |
| 23 | + } else { |
| 24 | + bucket.append((key, value)) |
| 25 | + } |
| 26 | + |
| 27 | + buckets[key % keySpace] = bucket |
| 28 | + } |
| 29 | + |
| 30 | + /** Returns the value to which the specified key is mapped, or -1 if this map contains no mapping for the key */ |
| 31 | + func get(_ key: Int) -> Int { |
| 32 | + let bucket = buckets[key % keySpace] |
| 33 | + |
| 34 | + if let index = find(key, bucket) { |
| 35 | + return bucket[index].1 |
| 36 | + } else { |
| 37 | + return -1 |
| 38 | + } |
| 39 | + } |
| 40 | + |
| 41 | + /** Removes the mapping of the specified value key if this map contains a mapping for the key */ |
| 42 | + func remove(_ key: Int) { |
| 43 | + var bucket = buckets[key % keySpace] |
| 44 | + |
| 45 | + guard let index = find(key, bucket) else { |
| 46 | + return |
| 47 | + } |
| 48 | + |
| 49 | + bucket.swapAt(index, bucket.count - 1) |
| 50 | + bucket.removeLast() |
| 51 | + |
| 52 | + buckets[key % keySpace] = bucket |
| 53 | + } |
| 54 | + |
| 55 | + private func find(_ key: Int, _ bucket: [(Int, Int)]) -> Int? { |
| 56 | + for (i, pair) in bucket.enumerated() where pair.0 == key { |
| 57 | + return i |
| 58 | + } |
| 59 | + |
| 60 | + return nil |
| 61 | + } |
| 62 | +} |
0 commit comments