让我们来看LeetCode上的这个实例
可以看出,其本质就是交换每个结点的左右子树,那么既然是这样的话,就需要遍历到二叉树的每个结点。则你选择前序遍历、中序遍历、后序遍历、层序遍历。都可以。只要是可以遍历到二叉树的每个结点,然后交换它的左右子树
以下是中序遍历实现翻转的代码:
public TreeNode invertTree(TreeNode root){
if (root == null) return null;
invertTree(root.left);
TreeNode tmp = root.left;
root.left = root.right;
root.right = tmp;
invertTree(root.left);
//注意上面中序遍历处理当前结点的时候左右子树发生了交换,所以这列写root.left其实是结点交换前的right
// Queue<TreeNode> queue = new LinkedList<>();
// queue.offer(root);
// while(!queue.isEmpty()){
// TreeNode node = queue.poll();
// TreeNode temp = node.left;
// node.left = node.right;
// node.right = temp;
// if (node.left != null) queue.offer(node.left);
// if (node.right != null) queue.offer(node.right);
// }
return root;
}
以下给出实现二叉树翻转的层序遍历的代码:
//翻转二叉树
public TreeNode invertTree(TreeNode root){
if (root == null) return null;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while(!queue.isEmpty()){
TreeNode node = queue.poll();
TreeNode temp = node.left;
node.left = node.right;
node.right = temp;
if (node.left != null) queue.offer(node.left);
if (node.right != null) queue.offer(node.right);
}
return root;
}
请读者自行实现前序,后序翻转二叉树代码。若需要帮助请在评论区留言!!!