Aktualizovat u7/hash_table.py
This commit is contained in:
parent
fa20d2ae88
commit
375babbc41
113
u7/hash_table.py
113
u7/hash_table.py
@ -15,24 +15,27 @@ class HashTable:
|
||||
def capacity(self):
|
||||
return self._capacity
|
||||
|
||||
def hash(self, key):
|
||||
def _validate_key(self, key):
|
||||
if not isinstance(key, str):
|
||||
raise TypeError("Key must be a string")
|
||||
|
||||
def hash(self, key):
|
||||
self._validate_key(key)
|
||||
|
||||
h = 0
|
||||
for znak in key:
|
||||
h = (31 * h + ord(znak)) % self.capacity
|
||||
for char in key:
|
||||
h = (31 * h + ord(char)) % self.capacity
|
||||
|
||||
return h
|
||||
|
||||
def probe(self, key):
|
||||
if not isinstance(key, str):
|
||||
raise TypeError("Key must be a string")
|
||||
self._validate_key(key)
|
||||
|
||||
start = self.hash(key)
|
||||
first_deleted = None
|
||||
|
||||
for krok in range(self.capacity):
|
||||
index = (start + krok) % self.capacity
|
||||
for step in range(self.capacity):
|
||||
index = (start + step) % self.capacity
|
||||
item = self._table[index]
|
||||
|
||||
if item is None:
|
||||
@ -52,87 +55,52 @@ class HashTable:
|
||||
raise RuntimeError("Hash table is full")
|
||||
|
||||
def __setitem__(self, key, value):
|
||||
if not isinstance(key, str):
|
||||
raise TypeError("Key must be a string")
|
||||
|
||||
try:
|
||||
index = self.probe(key)
|
||||
except RuntimeError:
|
||||
index = None
|
||||
|
||||
if index is not None:
|
||||
item = self._table[index]
|
||||
if item is not None and item is not self._DELETED:
|
||||
if item[0] == key:
|
||||
self._table[index] = (key, value)
|
||||
return
|
||||
|
||||
if (self._size + 1) / self.capacity > 0.7:
|
||||
old_table = self._table
|
||||
|
||||
self._capacity *= 2
|
||||
self._table = [None] * self._capacity
|
||||
self._size = 0
|
||||
|
||||
for item in old_table:
|
||||
if item is not None and item is not self._DELETED:
|
||||
old_key, old_value = item
|
||||
new_index = self.probe(old_key)
|
||||
self._table[new_index] = (old_key, old_value)
|
||||
self._size += 1
|
||||
self._validate_key(key)
|
||||
|
||||
index = self.probe(key)
|
||||
item = self._table[index]
|
||||
|
||||
# Existujúci kľúč: prepíš hodnotu.
|
||||
if item is not None and item is not self._DELETED:
|
||||
if item[0] == key:
|
||||
self._table[index] = (key, value)
|
||||
return
|
||||
|
||||
# Nový kľúč: skontroluj zaplnenie.
|
||||
if (self._size + 1) / self.capacity > 0.7:
|
||||
self._resize(self.capacity * 2)
|
||||
index = self.probe(key)
|
||||
|
||||
self._table[index] = (key, value)
|
||||
self._size += 1
|
||||
|
||||
def __getitem__(self, key):
|
||||
if not isinstance(key, str):
|
||||
raise TypeError("Key must be a string")
|
||||
|
||||
try:
|
||||
index = self.probe(key)
|
||||
except RuntimeError:
|
||||
raise KeyError(key)
|
||||
self._validate_key(key)
|
||||
|
||||
index = self.probe(key)
|
||||
item = self._table[index]
|
||||
|
||||
if item is None or item is self._DELETED:
|
||||
raise KeyError(key)
|
||||
|
||||
if item[0] != key:
|
||||
if item is None or item is self._DELETED or item[0] != key:
|
||||
raise KeyError(key)
|
||||
|
||||
return item[1]
|
||||
|
||||
def __delitem__(self, key):
|
||||
if not isinstance(key, str):
|
||||
raise TypeError("Key must be a string")
|
||||
|
||||
try:
|
||||
index = self.probe(key)
|
||||
except RuntimeError:
|
||||
raise KeyError(key)
|
||||
self._validate_key(key)
|
||||
|
||||
index = self.probe(key)
|
||||
item = self._table[index]
|
||||
|
||||
if item is None or item is self._DELETED:
|
||||
raise KeyError(key)
|
||||
|
||||
if item[0] != key:
|
||||
if item is None or item is self._DELETED or item[0] != key:
|
||||
raise KeyError(key)
|
||||
|
||||
self._table[index] = self._DELETED
|
||||
self._size -= 1
|
||||
|
||||
def __contains__(self, key):
|
||||
if not isinstance(key, str):
|
||||
raise TypeError("Key must be a string")
|
||||
|
||||
try:
|
||||
index = self.probe(key)
|
||||
except RuntimeError:
|
||||
return False
|
||||
self._validate_key(key)
|
||||
|
||||
index = self.probe(key)
|
||||
item = self._table[index]
|
||||
|
||||
return (
|
||||
@ -148,3 +116,20 @@ class HashTable:
|
||||
for item in self._table:
|
||||
if item is not None and item is not self._DELETED:
|
||||
yield item[0]
|
||||
|
||||
def load_factor(self):
|
||||
return self._size / self.capacity
|
||||
|
||||
def _resize(self, new_capacity):
|
||||
old_table = self._table
|
||||
|
||||
self._capacity = new_capacity
|
||||
self._table = [None] * new_capacity
|
||||
self._size = 0
|
||||
|
||||
for item in old_table:
|
||||
if item is not None and item is not self._DELETED:
|
||||
key, value = item
|
||||
index = self.probe(key)
|
||||
self._table[index] = (key, value)
|
||||
self._size += 1
|
||||
Loading…
Reference in New Issue
Block a user