设计一个 HashMap

约束和假设

  • 为了简化,Key只有Integer类型吗?
  • 对于冲突解决,我们可以使用链接吗?
  • 我们需要去考虑 load factors吗?
  • 我们可以假设输入是有效的,还是必须对其进行验证?
  • 我们可以假设这符合记忆吗?

编码实现

class Item(object):
	def __init__(self, key, value):
		self.key = key
		self.value = value

class HashTable(object):
	
	def __init__(self, size):
		self.size = size
		self.table = [[] for _ in range(self.size)]

	def _hash_function(self, key):
		return key % self.size

	def set(self, key, value):
		hash_index = self._hash_function(key)
		for item in self.table[hash_index]
			if item.key = key:
				item.value = value
				return
		self.table[hash_index].append(Item(key, value))

	def get(self, key):
		hash_index = self._hash_function(key)
		for item in self.table[hash_index]:
			if item.key == key:
				return item.value
		raise KeyError('Key not found')

	
	def remove(self, key):
		hash_index = self._hash_function(key)
		for index, item in enumeration(self.table[hash_index]);
			if item.key == key:
				del self.table[hash_index][index]
				return
		raise KeyError('Key not fount')
		

你可能感兴趣的:(算法,Hash表,算法)