编写递归算法,将二叉树中所有结点的左右右子 树相互交换。
时间: 2024-05-16 17:17:56 浏览: 70
可以按照以下步骤编写递归算法:
1. 首先检查根节点是否为空,如果为空则返回。
2. 交换当前节点的左右子树。
3. 递归调用函数,分别传入当前节点的左子树和右子树。
下面是示例代码:
```python
def swap_tree(root):
if root is None:
return
# 交换左右子树
root.left, root.right = root.right, root.left
# 递归处理左右子树
swap_tree(root.left)
swap_tree(root.right)
```
可以通过以下方式测试算法:
```python
# 定义二叉树节点类
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
# 构造二叉树
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.right.left = TreeNode(6)
root.right.right = TreeNode(7)
# 打印交换前的中序遍历结果
inorder_before = []
def inorder_traversal(root):
if root is None:
return
inorder_traversal(root.left)
inorder_before.append(root.val)
inorder_traversal(root.right)
inorder_traversal(root)
print("交换前的中序遍历结果:", inorder_before)
# 执行交换函数
swap_tree(root)
# 打印交换后的中序遍历结果
inorder_after = []
inorder_traversal(root)
print("交换后的中序遍历结果:", inorder_after)
```
输出结果应该为:
```
交换前的中序遍历结果: [4, 2, 5, 1, 6, 3, 7]
交换后的中序遍历结果: [7, 3, 6, 1, 5, 2, 4]
```
阅读全文