题目:
给你一棵二叉树的根节点 root ,翻转这棵二叉树,并返回其根节点。
思路:
对于叶子节点,不做考虑,直接continue。对于非叶子节点,如果其有两个孩子节点,将这两个孩子节点互换。之后将这两个孩子节点加入队列,以便后续考虑。如果只有一个孩子节点,假设有左孩子节点,将其作为右孩子节点,原左孩子节点设为null。
第一个continue是为了不在对叶子节点做多余判断。
第二个continue,是因为我们将node的左、右孩子节点已经加入了队列,不在执行下面代码,防止重复加入队列。
第三个continue,判断条件为node有左孩子,我们可知,其必没有右孩子节点,正常思路应该不执行
if(node.right != null){
node.left = node.right;
node.right = null;
queue.add(node.left);
}
但是我们对换了左孩子和右孩子,此时右孩子不为null,所以此时直接跳过,以防执行下面的语句。
完整代码如下:
class Solution {
public TreeNode invertTree(TreeNode root) {
if(root == null){
return null;
}
Queue<TreeNode> queue = new LinkedList<>();
TreeNode node = null;
queue.add(root);
while(!queue.isEmpty()){
node = queue.poll();
if(node.left == null && node.right == null){
continue;
}
if(node.left != null && node.right != null){
TreeNode temp = node.left;
node.left = node.right;
node.right = temp;
queue.add(node.left);
queue.add(node.right);
continue;
}
//root.right必是null
if(node.left != null){
node.right = node.left;
node.left = null;
queue.add(node.right);
continue;
}
if(node.right != null){
node.left = node.right;
node.right = null;
queue.add(node.left);
}
}
return root;
}
}