在二分搜索树中查找2个节点的路径

时间:2018-10-19 03:44:38

标签: python binary-search-tree nodes

我有这段代码可以找到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

0 个答案:

没有答案