牛客题霸NC72二叉树的镜像Java题解
方法1:利用栈
解题思路:利用栈遍历二叉树的所有节点,交换每个节点的左右子节点。
import java.util.*;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public void Mirror(TreeNode root) {
if(root == null){ //如果roto为null,返回
return;
}
LinkedList<TreeNode> stack = new LinkedList<>();
stack.addLast(root);
while(!stack.isEmpty()){ //利用栈遍历二叉树的所有节点,交换每个节点的左右子节点。
TreeNode node = stack.removeLast();
if(node.left!=null){ //如果左子节点不为null,将左子节点加入到stack中
stack.addLast(node.left);
}
if(node.right!=null){ //如果右子节点不为null,将右子节点加入到stack中
stack.addLast(node.right);
}
TreeNode tmp = node.left; //交换左、右子节点
node.left = node.right;
node.right = tmp;
}
}
}
方法2:递归
解题思路:递归遍历左右子树,交换每个节点的左右子节点。
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public void Mirror(TreeNode root) {
if (root == null) {
return;
}
Mirror(root.left); //递归遍历左子树
Mirror(root.right); //递归遍历右子树
TreeNode node = root.left; //交换左、右子节点
root.left = root.right;
root.right = node;
}
}