怎么用python实现二叉树搜索
时间: 2023-10-02 14:03:43 浏览: 99
Python实现二叉搜索树
5星 · 资源好评率100%
二叉树搜索可以通过以下Python代码实现:
```
# 定义二叉树结构
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
# 二叉树搜索函数
def searchBST(root, val):
if not root or root.val == val:
return root
elif root.val > val:
return searchBST(root.left, val)
else:
return searchBST(root.right, val)
```
其中,`TreeNode`类表示二叉树的结构,`searchBST`函数实现了二叉树的搜索功能,输入参数为根节点及待搜索的值。函数通过比较当前节点值与待搜索值的大小关系,递归地在左子树或右子树中继续搜索,直到找到目标节点或到达空节点为止。
阅读全文