Python搜尋樹

二元搜尋樹(BST)是一棵樹，其所有節點都遵循下述屬性 - 節點的左子樹的鍵小於或等於其父節點的鍵。 節點的右子樹的鍵大於其父節點的鍵。 因此，BST將其所有子樹分成兩部分; 左邊的子樹和右邊的子樹，可以定義為 -

left_subtree (keys)  ≤  node (key)  ≤  right_subtree (keys)

在B樹搜尋的值

在樹中搜尋值涉及比較輸入值與退出節點的值。 在這裡，也從左到右遍歷節點，最後是父節點。 如果搜尋到的值與任何退出值都不匹配，則返回未找到的訊息，否則返回找到的訊息。

class Node:

    def __init__(self, data):

        self.left = None
        self.right = None
        self.data = data

# Insert method to create nodes
    def insert(self, data):

        if self.data:
            if data < self.data:
                if self.left is None:
                    self.left = Node(data)
                else:
                    self.left.insert(data)
            elif data > self.data:
                if self.right is None:
                    self.right = Node(data)
                else:
                    self.right.insert(data)
        else:
            self.data = data
# findval method to compare the value with nodes
    def findval(self, lkpval):
        if lkpval < self.data:
            if self.left is None:
                return str(lkpval)+" Not Found"
            return self.left.findval(lkpval)
        elif lkpval > self.data:
            if self.right is None:
                return str(lkpval)+" Not Found"
            return self.right.findval(lkpval)
        else:
            print(str(self.data) + ' is found')
# Print the tree
    def PrintTree(self):
        if self.left:
            self.left.PrintTree()
        print( self.data),
        if self.right:
            self.right.PrintTree()


root = Node(12)
root.insert(6)
root.insert(14)
root.insert(3)
print(root.findval(7))
print(root.findval(14))

執行上面範例程式碼，得到以下結果 -

7 Not Found
14 is found
None