-
Notifications
You must be signed in to change notification settings - Fork 0
/
100_same_tree.py
67 lines (49 loc) · 1.4 KB
/
100_same_tree.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
'''
100. Same Tree
Given two binary trees, write a function to check if they are the same or not.
Two binary trees are considered the same if they are structurally identical and the nodes have the same value.
Example 1:
Input: 1 1
/ \ / \
2 3 2 3
[1,2,3], [1,2,3]
Output: true
Example 2:
Input: 1 1
/ \
2 2
[1,2], [1,null,2]
Output: false
Example 3:
Input: 1 1
/ \ / \
2 1 1 2
[1,2,1], [1,1,2]
Output: false
def isSameTree(self, p, q):
'''
'''
approach:
recursion
TC: O(N), SC: O(N) worst case O(log(N)) if completely balanced
'''
class Solution(object):
def isSameTree(self, p, q):
if p and q:
return p.val == q.val and self.isSameTree(p.left, q.left) and self.isSameTree(p.right, q.right)
return p is q
# returns True if p==None and q==None else False
# lengthier
# p and q are both none
if not p and not q:
return True
# one of p and q is None
if not p or not q:
return False
if p.val != q.val:
return False
return self.isSameTree(p.right, q.right) and self.isSameTree(p.left, q.left)
if __name__ == '__main__':
# begin
s = Solution()
print(s.isSameTree([1, 2, 3], [1, 2, 3]))