剑指 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; } }