以二叉链表作为二叉树的存储结构,编写交换二叉树每个结点的左孩子和右孩子的算法
时间: 2023-11-14 15:06:48 浏览: 78
编写算法交换二叉树中所有结点的左右子树.doc
5星 · 资源好评率100%
交换二叉树每个节点的左孩子和右孩子的算法可以通过递归实现。具体步骤如下:
1. 如果当前节点为空,直接返回。
2. 交换当前节点的左右孩子节点。
3. 递归处理当前节点的左子树和右子树。
以下是具体实现的代码:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def invertTree(root: TreeNode) -> TreeNode:
if not root:
return None
# 交换左右子树
root.left, root.right = root.right, root.left
# 递归左右子树
invertTree(root.left)
invertTree(root.right)
return root
```
在上述代码中,我们定义了一个 `TreeNode` 类来表示二叉树中的每个节点。然后,我们定义了一个名为 `invertTree` 的函数,该函数接受一个二叉树的根节点,并返回交换后的根节点。
在函数中,我们首先检查当前节点是否为空。如果为空,我们直接返回。否则,我们交换当前节点的左右孩子节点,并递归处理当前节点的左子树和右子树。
最后,我们返回交换后的根节点。
阅读全文