226翻转二叉树
给你一棵二叉树的根节点 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
思路
这道题用递归的思想就是将根节点的左右儿子交换,然后再对子节点进行递归操作,直到子节点均为空。
但是我感觉那个TreeNode类型的函数拿来做递归有点奇怪,所以另外定义了一个递归函数。
class Solution {
public TreeNode invertTree(TreeNode root) {
if(root==null)
return null;
invert(root);
return root;
}
public void invert(TreeNode node){
if(node.left!=null||node.right!=null){
TreeNode tmp=node.left;
node.left=node.right;
node.right=tmp;
if(node.left!=null) invert(node.left);
if(node.right!=null) invert(node.right);
}
}
}
然后看了一下题解的递归,只用了一个函数就行了,只要把递归的结果作为交换过的节点值存储起来即可。
class Solution {
public TreeNode invertTree(TreeNode root) {
if (root == null) {
return null;
}
TreeNode left = invertTree(root.left);
TreeNode right = invertTree(root.right);
root.left = right;
root.right = left;
return root;
}
}
还有一种方法是层序遍历的同时将子节点交换。
public TreeNode invertTreeByQueue(TreeNode root) {
if (root == null) {
return null;
}
Queue<TreeNode> queue = new ArrayDeque<>();
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;
}
总结
这些方法实际上都使用了二叉树的前序、后序遍历法,可以用递归,层序遍历等方法实现。