Skip to content

Commit d3856c1

Browse files
authored
Merge pull request #1105 from ivan1016017/november27
adding invert binary tree
2 parents d7f1a61 + 5cc95fb commit d3856c1

File tree

2 files changed

+42
-0
lines changed

2 files changed

+42
-0
lines changed
Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
from typing import List, Union, Collection, Mapping, Optional
2+
from abc import ABC, abstractmethod
3+
4+
class TreeNode:
5+
def __init__(self, val=0, left=None, right=None):
6+
self.val = val
7+
self.left = left
8+
self.right = right
9+
10+
class Solution:
11+
def invertTree(self, root: TreeNode) -> TreeNode:
12+
13+
try:
14+
root.val
15+
except:
16+
return root
17+
18+
root.left, root.right = (
19+
self.invertTree(root.right),
20+
self.invertTree(root.left)
21+
)
22+
23+
return root
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
import unittest
2+
from src.my_project.interviews.top_150_questions_round_11\
3+
.invert_binary_tree import TreeNode, Solution
4+
5+
class InvertTreeTestCase(unittest.TestCase):
6+
7+
def test_none_inverted_tree(self):
8+
solution = Solution()
9+
tree = None
10+
output = solution.invertTree(root=tree)
11+
self.assertIsNone(output)
12+
13+
def test_inverted_tree(self):
14+
solution = Solution()
15+
tree = TreeNode(1,TreeNode(2),TreeNode(3))
16+
output = solution.invertTree(root=tree)
17+
self.assertEqual(1,output.val)
18+
self.assertEqual(2,output.right.val)
19+
self.assertEqual(3,output.left.val)

0 commit comments

Comments
 (0)