Skip to content

Commit e94e77c

Browse files
authored
Merge pull request #1279 from ivan1016017/may23
adding algo
2 parents 3712b81 + b7e928a commit e94e77c

File tree

2 files changed

+37
-0
lines changed

2 files changed

+37
-0
lines changed
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
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 isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
12+
13+
if p and q:
14+
return p.val == q.val \
15+
and self.isSameTree(p.left, q.left) \
16+
and self.isSameTree(p.right, q.right)
17+
else:
18+
return p is q
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_16\
3+
.same_tree import Solution, TreeNode
4+
5+
class SameTreeTestCase(unittest.TestCase):
6+
7+
def test_is_same_tree(self):
8+
solution = Solution()
9+
tree1 = TreeNode(1, TreeNode(2), TreeNode(3))
10+
tree2 = TreeNode(1, TreeNode(2), TreeNode(3))
11+
output = solution.isSameTree(p=tree1, q=tree2)
12+
return self.assertTrue(output)
13+
14+
def test_is_no_same_tree(self):
15+
solution = Solution()
16+
tree1 = TreeNode(1, TreeNode(2), TreeNode(3))
17+
tree2 = TreeNode(1, TreeNode(3), TreeNode(2))
18+
output = solution.isSameTree(p=tree1, q=tree2)
19+
return self.assertFalse(output)

0 commit comments

Comments
 (0)