我有这段代码可以找到2个节点之间的路径。问题是,如果节点2大于节点1,则无法区分是父节点还是正子,现在我要解决这个问题。我认为也许将element2和1交换会起作用,但这只会使顺序错误。
def _find_path(self, element_1, element_2, nature):
if element_1 !=None and element_2 != None:
if element_1.key == element_2.key:
self.list_path.append(element_1)
else:
if(element_1.key == self.root.key or element_2.key == self.root.key):
if(element_2.key > element_1.key):
self.list_path.append(element_1)
self._find_path(element_1.rightChild, element_2, nature)
if (element_2.key < element_1.key):
self.list_path.append(element_1)
self._find_path(element_1.leftChild, element_2, nature)
else:
if(element_1.key < self.root.key and element_2.key < self.root.key or element_1.key > self.root.key and element_2.key > self.root.key):
if element_2.key > element_1.key:
self.list_path.append(element_1)
self._find_path(element_1.parent, element_2, nature)
if element_2.key < element_1.key:
self.list_path.append(element_1)
self._find_path(element_1.leftChild, element_2, nature)
else:
if element_1.key == element_2.key:
return [element_1]
res = self._find_path(element_1.leftChild, element_2,nature)
if res:
self.steps += 1
return [element_1] + res
res = self._find_path(element_1.rightChild, element_2,nature)
if res:
self.steps += 1
return [element_1] + res
else:
if(element_1.key < self.root.key and element_2.key > self.root.key):
self.list_path.append(element_1)
self._find_path(element_1.parent, element_2, nature)
else:
self.list_path.append(element_1)
self._find_path(element_1.parent, element_2, nature)
return self.list_path