二叉排序树python的非递归搜索
时间: 2024-06-04 11:04:53 浏览: 88
二叉排序树的非递归搜索可以通过迭代实现。具体步骤如下:
1. 从根节点开始,将根节点入栈。
2. 循环执行以下步骤,直到栈为空或者找到目标节点为止:
a. 取出栈顶节点,判断该节点是否为目标节点,若是则返回该节点。
b. 若该节点不是目标节点,则将其右子节点和左子节点(如果存在)入栈。
3. 若栈为空,说明未找到目标节点,返回空。
下面是示例代码:
```python
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
def search(root, target):
if not root:
return None
stack = [root]
while stack:
node = stack.pop()
if node.val == target:
return node
if node.right:
stack.append(node.right)
if node.left:
stack.append(node.left)
return None
```
相关问题
二叉排序树的非递归查找
二叉排序树的非递归查找可以通过循环来实现。具体步骤如下:
1. 从根节点开始,将要查找的值与当前节点的值进行比较。
2. 如果要查找的值等于当前节点的值,则找到了目标节点,返回该节点。
3. 如果要查找的值小于当前节点的值,则将当前节点指向其左子节点,并继续比较。
4. 如果要查找的值大于当前节点的值,则将当前节点指向其右子节点,并继续比较。
5. 如果当前节点为空,则表示未找到目标节点,返回空。
以下是一个示例代码来演示二叉排序树的非递归查找:
```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):
while root:
if val == root.val:
return root
elif val < root.val:
root = root.left
else:
root = root.right
return None
# 创建一个二叉排序树
root = TreeNode(4)
root.left = TreeNode(2)
root.right = TreeNode(7)
root.left.left = TreeNode(1)
root.left.right = TreeNode(3)
# 在二叉排序树中查找值为2的节点
target = searchBST(root, 2)
if target:
print("找到了目标节点,值为:", target.val)
else:
print("未找到目标节点")
```
输出结果为:
```
找到了目标节点,值为: 2
```
二叉排序树的非递归建立
二叉排序树的非递归建立可以使用栈来实现。具体步骤如下:
1. 创建一个空树,将第一个节点作为根节点。
2. 从第二个节点开始,依次将节点插入到树中。
3. 对于每个待插入的节点,从根节点开始遍历树,如果该节点的值小于当前节点的值,则继续遍历当前节点的左子树;如果该节点的值大于当前节点的值,则继续遍历当前节点的右子树。
4. 找到待插入节点应该插入的位置后,将其插入到该位置,并将其父节点入栈。
5. 重复步骤3和4,直到所有节点都被插入到树中。
以下是具体实现代码:
```python
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
def build_bst(nums):
if not nums:
return None
root = TreeNode(nums[0])
stack = [root]
for num in nums[1:]:
node = TreeNode(num)
parent = None
while stack:
if num < stack[-1].val:
parent = stack[-1]
stack.pop()
else:
break
if parent:
parent.left = node
else:
stack[-1].right = node
stack.append(node)
return root
```