给你一棵二叉树的根节点 root
,翻转这棵二叉树,并返回其根节点。
示例 1:
输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]
示例 2:
输入:root = [2,1,3]
输出:[2,3,1]
示例 3:
输入:root = []
输出:[]
提示:
- 树中节点数目范围在
[0, 100]
内 -100 <= Node.val <= 100
解题思路:
遍历每个节点,互换他们的左右子树即可。
代码实现:
class Solution {
public TreeNode invertTree(TreeNode root) {
//为空则返回
if(root==null)return root;
//进入索引
bfs(root);
return root;
}
public void bfs(TreeNode root){
//树不为空则翻转
if(root!=null){
//左右子树都有
if(root.left!=null&&root.right!=null){
//左右子树互换
TreeNode p=root.right;
root.right=root.left;
root.left=p;
bfs(root.left);
bfs(root.right);
}
//只有左子树
else if(root.left!=null){
//左子树放到右子树上
root.right=root.left;
root.left=null;
bfs(root.right);
}
//只有右子树
else if(root.right!=null){
//右子树放到左子树上
root.left=root.right;
root.right=null;
bfs(root.left);
}
}
}
}