1. 题目描述
Invert a binary tree.
4
/ \
2 7
/ \ / \
1 3 6 9
to
4
/ \
7 2
/ \ / \
9 6 3 1
翻转一个二叉树,即交换二叉树所有的左右节点。
2. 解题思路
树的深度优先访问,交换左右节点后访问孩子节点,交换孩子的左右节点。递归时首先以一个局部作为递归过程中的节点。如选取上例中的节点2,操作到这个节点时,首先需要判断这个节点是否是空(出口),这样做就避免了在递归过程中判断左右孩子是否为空的过程。之后,判断完这个节点是否为空后查看这个节点是否有孩子(出口),若没有孩子也就没有必要交换了,如叶节点3。第三步,交换这个节点的左右孩子,交换这个节点的孩子下的孩子(递归的过程)。
3. Code
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode invertTree(TreeNode root) {
// 节点为空时返回空
if(root == null)
{
return null;
}
// 节点左右都为空时返回当前节点
else if(root.left == null && root.right == null)
{
return root;
}
// 节点左右节点有一个是空或都不为空
else
{
// 交换左右节点
TreeNode tmp = root.left;
root.left = root.right;
root.right = tmp;
// 交换左节点下的孩子
invertTree(root.left);
// 交换右节点下的孩子
invertTree(root.right);
return root;
}
}
}