python实现一个简易hashmap

py3study

发布于 2020-01-07 11:28:21

1.1K0

发布于 2020-01-07 11:28:21

文章被收录于专栏：python3python3

python实现一个简易hashmap,不严谨、有问题之处请多多指出。。

近日把数据结构翻出来看看，发现自己这方面的知识很欠缺，算是自己的记录，也希望给正在学习数据结构的老铁们分享，共同学习。。。

简单说明原理 python语言中的dict底层是基于hashmap结构实现的，dict的使用就不说了。关键一点是，hashmap可以在一堆数据中，很快的根据key，找到value，这个关键点主要是由hash函数实现的。详细原理请看《大话数据结构》一书的8.9章节，我觉得讲得很好。。

简单实现 《大话数据结构》结构一书中主要用C语言来实现hashmap结构，下面我会给出用python语言实现的代码。并且为解决hash冲突问题，我使用了“链地址法”的结构。 MyHash内部使用items列表来存储数据，items是一个列表，并且每个元素也是一个列表，元素列表中存储了具体的（key,value）元组，不同的key根据hash函数先算出index，即存储在哪条列表中，插入时则直接append，查找时则根据equals方法将待查找的key与列表中的所有元组的第一个值（key）进行比较，找到相等的则返回元组的第二个值（value），找不到则raise KeyError异常。

# coding=utf-8


class MyHash(object):

    def __init__(self, length=10):
        self.length = length
        self.items = [[] for i in range(self.length)]

    def hash(self, key):
        """计算该key在items哪个list中，针对不同类型的key需重新实现"""
        return key % self.length

    def equals(self, key1, key2):
        """比较两个key是否相等，针对不同类型的key需重新实现"""
        return key1 == key2

    def insert(self, key, value):
        index = self.hash(key)
        if self.items[index]:
            for item in self.items[index]:
                if self.equals(key, item[0]):
                    # 添加时若有已存在的key，则先删除再添加（更新value）
                    self.items[index].remove(item)
                    break
        self.items[index].append((key, value))
        return True

    def get(self, key):
        index = self.hash(key)
        if self.items[index]:
            for item in self.items[index]:
                if self.equals(key, item[0]):
                    return item[1]
        # 找不到key，则抛出KeyError异常
        raise KeyError

    def __setitem__(self, key, value):
        """支持以 myhash[1] = 30000 方式添加"""
        return self.insert(key, value)

    def __getitem__(self, key):
        """支持以 myhash[1] 方式读取"""    
        return self.get(key)


myhash = MyHash()
myhash[1] = 30000
myhash.insert(2, 2100)
print myhash.get(1)
myhash.insert(1, 3)
print myhash.get(2)
print myhash.get(1)
print myhash[1]

几点说明