剑指 Offer 27. 二叉树的镜像(递归+迭代解法)
问题描述
请完成一个函数,输入一个二叉树,该函数输出它的镜像。 例如输入: 4 / 2 7 / / 1 3 6 9 镜像输出: 4 / 7 2 / / 9 6 3 1 示例 1: 输入:root = [4,2,7,1,3,6,9] 输出:[4,7,2,9,6,3,1]
解题思路
解法一:递归
这道题即翻转二叉树。
使用递归解决是最快的。
将当前节点的左右节点互换。
再交换左节点。
交换右节点。
到达叶子节点即为结束。
由于需要完整的递归整颗二叉树,所以不需要返回值。
解法二:栈(迭代)
将每一层的节点放入栈中,交换左右节点。到达叶子节点为止。
Java解法
解法一(递归):
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode mirrorTree(TreeNode root) {
if(root==null) return root;
helper(root);
return root;
}
public void helper(TreeNode node){
if(node!=null){
TreeNode tmp = node.left;
node.left = node.right;
node.right = tmp;
}else{
return;
}
helper(node.left);
helper(node.right);
}
}
解法二(迭代——栈)
class Solution {
public TreeNode mirrorTree(TreeNode root) {
if(root==null) return root;
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while(!stack.empty()){
TreeNode node = stack.pop();
if(node!=null){
TreeNode tmp = node.left;
node.left = node.right;
node.right = tmp;
stack.push(node.left);
stack.push(node.right);
}
}
return root;
}
}
