我尝试用Python实现跳表,可以帮我吗?很简单
这段代码非常简单,我的参考资料来自这里:http://www.mathcs.emory.edu/~cheung/Courses/323/Syllabus/Map/skip-list-impl.html#why-q
我觉得插入功能还不错,但当我尝试使用 get()
函数时,它什么都不返回,反而在 searchEntry()
的部分无限循环。我不知道哪里出了问题。在 insert()
函数中,searchEntry()
的运行是正常的。它返回的是一个引用,指向一个包含比要插入的键小的键的 floorEntry(k)
条目。请帮我找出 searchEntry()
函数中的错误来源。我很抱歉我对此不是很在行。谢谢!
from QuadLinkedList import QLLNode
import random
class Skippy:
def __init__(self):
self._p1 = QLLNode("MINUS_INF")
self._p2 = QLLNode("PLUS_INF")
self._head = self._p1
self._tail = self._p2
self._p1.setNext(self._p2)
self._p2.setPrev(self._p1)
self._height = 0
self._n = 0
def insert(self, key, value):
p = self.searchEntry(key)
print "p = " + str(p.getKey())
q = QLLNode(key, value)
q.setPrev(p)
q.setNext(p.getNext())
p.getNext().setPrev(q)
p.setNext(q)
i = 0
while random.randint(0,1) != 0:
if i >= self._height:
self._height += 1
newHead = QLLNode("MINUS_INF")
newTail = QLLNode("PLUS_INF")
newHead.setNext(newTail)
newHead.setDown(self._head)
newTail.setPrev(newHead)
newTail.setDown(self._tail)
self._head.setUp(newHead)
self._tail.setUp(newTail)
self._head = newHead
self._tail = newTail
while p.getUp() == None:
p = p.getPrev()
p = p.getUp()
e = QLLNode(key,None)
e.setPrev(p)
e.setNext(p.getNext())
e.setDown(q)
p.getNext().setPrev(e)
p.setNext(e)
q.setUp(e)
q = e
i += 1
self._n += 1
return None
def get(self, key):
p = self.searchEntry(key)
if key == p.getKey():
return p.getElement()
else:
return "There's None!"
def searchEntry(self, key):
p = self._head
while True:
while p.getNext().getKey() != "PLUS_INF" and p.getNext().getKey() <= key:
p = p.getNext()
if p.getDown() != None:
p = p.getDown()
else:
break
return p
1 个回答
1
问题不在于searchEntry
的代码,逻辑看起来是正确的。真正的问题在于列表的结构被搞乱了。我觉得问题出在你在insert
中添加新层级的代码上。具体来说,就是这部分:
if i >= self._height: #make an empty level
self._height += 1
self._minus.setNext(self._plus)
self._minus.setDown(self._head)
self._plus.setPrev(self._minus)
self._plus.setDown(self._tail)
self._head.setUp(self._minus)
self._tail.setUp(self._plus)
self._head = self._minus
self._tail = self._plus
我注意到这段代码的一个问题是,你没有创建任何新的节点,只是在修改已有的节点,这样就破坏了你的列表结构。我认为你需要创建新的head
和tail
节点,并把它们链接到结构的顶部。(minus
和plus
在你提供的链接中的算法里并没有提到,所以我不太清楚你在用它们做什么。)可能你想要的是这样的:
if i >= self._height: #make an empty level
self._height += 1
newHead = QLLNode("MINUS_INF")
newTail = QLLNode("PLUS_INF")
newHead.setNext(newTail)
newHead.setDown(self._head)
newTail.setPrev(newHead)
newTail.setDown(self._tail)
self._head.setUp(newHead)
self._head = newHead
self._tail.setUp(newTail)
self._tail = newTail