Leetcode solution 226. Invert Binary Tree

Posted 2021-03-18 baozitraining

Problem Statement 

Invert a binary tree.

Example:

Input:

     4
   /     2     7
 /    / 1   3 6   9

Output:

     4
   /     7     2
 /    / 9   6 3   1

Trivia:

This problem was inspired by this original tweet by Max Howell:

Google: 90% of our engineers use the software you wrote (Homebrew), but you can’t invert a binary tree on a whiteboard so f*** off.

 Problem link

Video Tutorial

You can find the detailed video tutorial here

• Youtube
• B站

Thought Process

Simple question on understanding about recursion. It could be solved using pre-order and post-order traversal style recursion. It can also be solved iteratively using a queue. Please refer to Leetcode official solution. It‘s very similar to "Binary Tree ZigZag Level Order Traversal". Remember a full binary tree max leaf is (N+1)/2 or ceiling of N/2

, that‘s the O(N) space complexity using a queue.

 

If you are like Max Howell, you can show off but I would still recommend to stay humble if you really want the job. Brillient jerks are rare (IMHO Linus Torvalds is one) but very productive, but not all companies would accept those culture (Netflix does but not others). Be strong and humble.

Solutions

 1 public TreeNode invertTree(TreeNode root) {
 2     return this.helper(root);
 3 }
 4 
 5 // post order
 6 private TreeNode helper(TreeNode root) {
 7     if (root == null) {
 8         return root;
 9     }
10 
11     TreeNode l = helper(root.left);
12     TreeNode r = helper(root.right);
13 
14     root.left = r;
15     root.right = l;
16 
17     return root;
18 }
19 
20 // Preorder
21 private TreeNode invertHelper(TreeNode root) {
22     if (root == null) {
23         return null;
24     }
25 
26     TreeNode temp = root.left;
27     root.left = root.right;
28     root.right = temp;
29 
30     this.invertHelper(root.left);
31     this.invertHelper(root.right);
32 
33     return root;
34 } 

Recursion (pre-order and post-order)

Time Complexity: O(N), where N is the total number of tree nodes

Space Complexity: O(1) Or O(lgN) if you count the recursion function stack

 

References

• Leetcode official solution