程序员社区

LeetCode-226-翻转二叉树

LeetCode-226-翻转二叉树

226. 翻转二叉树

难度-简单

翻转一棵二叉树。

示例:

输入:

     4
   /   \
  2     7
 / \   / \
1   3 6   9

输出:

     4
   /   \
  7     2
 / \   / \
9   6 3   1

备注:
这个问题是受到 Max Howell 的 原问题 启发的 :

谷歌:我们90%的工程师使用您编写的软件(Homebrew),但是您却无法在面试时在白板上写出翻转二叉树这道题,这太糟糕了。

LeetCode-226-翻转二叉树插图
image-20210526171422934

可以使用先序遍历,中序遍历,后序遍历

每一个节点左右交换

  • 先序遍历
class Solution {
    public TreeNode invertTree(TreeNode root) {
        if (root == null) return root;
       
       TreeNode tmp = root.left;
       root.left = root.right;
       root.right = tmp;
       
      invertTree(root.left);
      invertTree(root.right);
      
      return root;
    }
}
LeetCode-226-翻转二叉树插图1
image-20210526172551939
  • 中序遍历
class Solution {
    public TreeNode invertTree(TreeNode root) {
       if (root == null) return root;
       
      invertTree(root.left);

       TreeNode tmp = root.left;
       root.left = root.right;
       root.right = tmp;

      invertTree(root.left);
      
      return root;
    }
}
LeetCode-226-翻转二叉树插图2
image-20210526172650394
  • 后序遍历
class Solution {
    public TreeNode invertTree(TreeNode root) {
       if (root == null) return root;
       
      invertTree(root.left);
      invertTree(root.right);
       
       TreeNode tmp = root.left;
       root.left = root.right;
       root.right = tmp;
      
      return root;
    }
}
LeetCode-226-翻转二叉树插图3
image-20210526172742093
  • 层次遍历(宽度优先遍历)
class Solution {
    public TreeNode invertTree(TreeNode root) {
      if (root == null) return root;
        
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            TreeNode tmp = node.left;
            node.left = node.right;
            node.right = tmp;
            
            if (node.left != null) {
                queue.offer(node.left);
            }
            
            if (node.right != null) {
                queue.offer(node.right);
            }
        }
        return root;
    }
}
LeetCode-226-翻转二叉树插图4
image-20210526172857043
参考二叉树的基本算法
赞(0) 打赏
未经允许不得转载:IDEA激活码 » LeetCode-226-翻转二叉树

一个分享Java & Python知识的社区