翻转一棵二叉树。
示例:
输入:
4 / \ 2 7 / \ / \ 1 3 6 9
输出:
4 / \ 7 2 / \ / \ 9 6 3 1
递归做法:
public TreeNode invertTree(TreeNode root) {
if(root==null){
return null;
}
if(root.left!=null || root.right!=null){
TreeNode temp=root.left;
root.left=root.right;
root.right=temp;
}
invertTree(root.left);
invertTree(root.right);
return root;
}
非递归做法
public TreeNode invertTree(TreeNode root){
//创建一个队列
LinkedList<TreeNode> queue=new LinkedList<>();
if(root==null){
return null;
}
queue.add(root);
while(queue.size() != 0){
TreeNode head=queue.removeFirst();
TreeNode temp=head.left;
head.left=head.right;
head.right=temp;
if(head.left!=null){
queue.add(head.left);
}
if(head.right!=null){
queue.add(head.right);
}
}
return root;
}