From 0f92fc245cec51cb3bb35bfc4c2aa790199a1b02 Mon Sep 17 00:00:00 2001 From: marina <138340846+bt3gl-cryptographer@users.noreply.github.com> Date: Mon, 7 Aug 2023 16:37:51 -0700 Subject: [PATCH] Update hash_map_array.py --- hash_objects/hash_map_array.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/hash_objects/hash_map_array.py b/hash_objects/hash_map_array.py index d2f954b..6ec0031 100644 --- a/hash_objects/hash_map_array.py +++ b/hash_objects/hash_map_array.py @@ -36,19 +36,22 @@ class Bucket: class HashMap: def __init__(self, key_space): - self.key_space = key_space - self.table = [Bucket() for _ in range(self.key_space)] + self.size = size + self.table = [Bucket() for _ in range(self.size)] + + def _get_hash_key(self, key): + return key % self.size def put(self, key: int, value: int): - hash_key = key % self.key_space + hash_key = self._get_hash_key(key) self.table[hash_key].put(key, value) def get(self, key: int): - hash_key = key % self.key_space + hash_key = self._get_hash_key(key) return self.table[hash_key].get(key) def remove(self, key: int): - hash_key = key % self.key_space + hash_key = self._get_hash_key(key) self.table[hash_key].remove(key)