Skip to content
33 changes: 33 additions & 0 deletions data_structures/hashing/hash_map.py
Original file line number Diff line number Diff line change
Expand Up @@ -242,6 +242,25 @@ def __delitem__(self, key: KEY) -> None:
self._size_down()

def __getitem__(self, key: KEY) -> VAL:
"""
Returns the item at the given key

>>> hm = HashMap(5)
>>> hm._add_item(1, 10)
>>> hm.__getitem__(1)
10

>>> hm = HashMap(5)
>>> hm._add_item(10, -10)
>>> hm._add_item(20, -20)
>>> hm.__getitem__(20)
-20

>>> hm = HashMap(5)
>>> hm._add_item(-1, 10)
>>> hm.__getitem__(-1)
10
"""
for ind in self._iterate_buckets(key):
item = self._buckets[ind]
if item is None:
Expand All @@ -253,6 +272,20 @@ def __getitem__(self, key: KEY) -> VAL:
raise KeyError(key)

def __len__(self) -> int:
"""
Returns the number of items present in hashmap

>>> hm = HashMap(5)
>>> hm._add_item(1, 10)
>>> hm._add_item(2, 20)
>>> hm._add_item(3, 30)
>>> hm.__len__()
3

>>> hm = HashMap(5)
>>> hm.__len__()
0
"""
return self._len

def __iter__(self) -> Iterator[KEY]:
Expand Down