python實(shí)現(xiàn)跳表SkipList的示例代碼
發(fā)布日期：2022-02-22 08:59 | 文章來(lái)源：CSDN
跳表
跳表，又叫做跳躍表、跳躍列表，在有序鏈表的基礎(chǔ)上增加了“跳躍”的功能，由William Pugh于1990年發(fā)布，設(shè)計(jì)的初衷是為了取代平衡樹(shù)（比如紅黑樹(shù)）。
Redis、LevelDB 都是著名的 Key-Value 數(shù)據(jù)庫(kù)，而Redis中 的 SortedSet、LevelDB 中的 MemTable 都用到了跳表。
對(duì)比平衡樹(shù)，跳表的實(shí)現(xiàn)和維護(hù)會(huì)更加簡(jiǎn)單，跳表的搜索、刪除、添加的平均時(shí)間復(fù)雜度是 O(logn)。
跳表的結(jié)構(gòu)如圖所示：
可以發(fā)現(xiàn)，對(duì)于一個(gè)節(jié)點(diǎn)Node，其含有多個(gè)next指針，不同索引的next分別代表不同層次的下一個(gè)節(jié)點(diǎn)，下次是節(jié)點(diǎn)類(lèi)Node的python定義：
class Node(): def __init__(self,key,value,level): ''' :param level:每個(gè)node對(duì)應(yīng)的nexts層數(shù)不同 ''' self.key=key self.value=value self.nexts=[None]*level#節(jié)點(diǎn)類(lèi)型next指針，初始值為空 def __str__(self): #return "[key:"+str(self.key)+", value:"+str(self.value)+" len:"+str(len(self.nexts))+"]" return "["+str(self.key)+","+str(self.value)+","+str(len(self.nexts))+"]"
關(guān)于添加、刪除、查找見(jiàn)一下完整代碼：
'''
跳表 Skip List ，其初衷是為了替代紅黑樹(shù)
'''
import random
import mkl_random
import time
class SkipList():
def __init__(self):
#頭節(jié)點(diǎn)不存儲(chǔ)任何數(shù)據(jù)
self.MAX_LEVEL = 32 # 最大level層數(shù)
self.__first=SkipList.Node(None, None, self.MAX_LEVEL)#頭節(jié)點(diǎn)
self.__level=0#實(shí)際的level層數(shù)
self.__size=0#Jiedian個(gè)數(shù)
self.__p=0.25#用于生成添加節(jié)點(diǎn)時(shí)的隨機(jī)level
return
class Node():
def __init__(self,key,value,level):
'''
:param level:每個(gè)node對(duì)應(yīng)的nexts層數(shù)不同
'''
self.key=key
self.value=value
self.nexts=[None]*level
def __str__(self):
#return "[key:"+str(self.key)+", value:"+str(self.value)+" len:"+str(len(self.nexts))+"]"
return "["+str(self.key)+","+str(self.value)+","+str(len(self.nexts))+"]"
def get(self,key):
'''
:param key:
:return: key對(duì)應(yīng)的value
'''
self.keyCheck(key)
node=self.__first
for level in range(self.__level - 1,-1,-1):
#在該層查找，key大于節(jié)點(diǎn)的key向前查找
while node.nexts[level] and node.nexts[level].key<key:
node=node.nexts[level]
if node.nexts[level] and node.nexts[level].key==key:#相等則找到，否則向下尋找
return node.nexts[level].value
return None
def put(self,key,value):
'''
return:原來(lái)的value，原來(lái)不存在key則為空
'''
self.keyCheck(key)
prev=[None]*self.__level
node=self.__first
for i in range(self.__level - 1, -1, -1):
while node.nexts[i] and node.nexts[i].key<key:
node=node.nexts[i]
if node.nexts[i] and node.nexts[i].key==key:
oldValue=node.nexts[i].value
node.nexts[i].value=value
return oldValue
prev[i]=node#保存當(dāng)前l(fā)evel小于key的node
newLevel=self.randomLevel()
newNode=SkipList.Node(key,value,newLevel)
for i in range(newLevel):
if i<self.__level:
newNode.nexts[i]=prev[i].nexts[i]
prev[i].nexts[i]=newNode
else:
self.__first.nexts[i]=newNode
self.__size+=1
self.__level=max(self.__level, newLevel)
return None
def remove(self,key):
'''
:return: 節(jié)點(diǎn)對(duì)應(yīng)的value值，不存在則返回None
'''
self.keyCheck(key)
prev=[None]*self.__level
node=self.__first
flag=False#該節(jié)點(diǎn)是否被查找到
for i in range(self.__level - 1, -1, -1):
while node.nexts[i] and node.nexts[i].key<key:
node=node.nexts[i]
if node.nexts[i].key==key:
flag=True
prev[i]=node
if not flag:
return None
removedNode=node.nexts[0]#需要被刪除的節(jié)點(diǎn)
for i in range(len(removedNode.nexts)):#該nexts一定小于等于prev的長(zhǎng)度
prev[i].next[i]=removedNode.nexts[i]
self.__size-=1
newLevel=self.__level
while newLevel>0 and not self.__first.nexts[newLevel - 1]:
newLevel-=1
self.__level=newLevel
return removedNode.value
def keyCheck(self, key):
'''
限制傳入key不能為空
'''
if key!=0 and not key:
raise AttributeError("key can not be None")
def size(self):
return self.__size
def isEmpty(self):
return self.__size == 0
def randomLevel(self):#生成一個(gè)隨機(jī)的層數(shù)
level=1
while mkl_random.rand()<self.__p and level<self.MAX_LEVEL:
level+=1
return level
def __str__(self):
result=""
for i in range(self.__level - 1, -1, -1):
result+=str(i)
node = self.__first
while node.nexts[i]:
result+=str(node.nexts[i])
node=node.nexts[i]
result+='\n'
print("level:"+str(self.__level))
return result
def showFirst(self):
for item in self.__first.nexts:
print(item,end=' ')
print()
def timeCalculate(container, size:int):
begin=time.time()
for i in range(size):
if isinstance(container,dict):
container[i]= i * 3
else:
container.put(i, i * 3)
error_count = 0
for i in range(size):
if container.get(i) != i * 3:
#print("wrong " + str(i) + ":" + str(skipList.get(i)))
error_count+=1
end=time.time()
print(type(container))
print(f'error rate:{float(error_count) / size:0.5f}')
print(f'time cost:{float(end-begin)*1000:0.3f} ms')
if __name__=='__main__':
timeCalculate({},1000000)
timeCalculate(SkipList(),10000)
到此這篇關(guān)于python實(shí)現(xiàn)跳表SkipList的文章就介紹到這了,更多相關(guān)python 跳表SkipList內(nèi)容請(qǐng)搜索本站以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持本站！
版權(quán)聲明：本站文章來(lái)源標(biāo)注為YINGSOO的內(nèi)容版權(quán)均為本站所有，歡迎引用、轉(zhuǎn)載，請(qǐng)保持原文完整并注明來(lái)源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站，禁止在非www.sddonglingsh.com所屬的服務(wù)器上建立鏡像，否則將依法追究法律責(zé)任。本站部分內(nèi)容來(lái)源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來(lái)，僅供學(xué)習(xí)參考，不代表本站立場(chǎng)，如有內(nèi)容涉嫌侵權(quán)，請(qǐng)聯(lián)系alex-e#qq.com處理。
相關(guān)文章
關(guān)注官方微信