forked from yingl/LintCodeInPython
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsubtree.py
More file actions
34 lines (32 loc) · 1.04 KB
/
Copy pathsubtree.py
File metadata and controls
34 lines (32 loc) · 1.04 KB
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
# -*- coding: utf-8 -*-
class Solution:
# @param T1, T2: The roots of binary tree.
# @return: True if T2 is a subtree of T1, or false.
def isSubtree(self, T1, T2):
# write your code here
if (T1 is None) and (T2 is None):
return True
elif T1 is None:
return False
elif T2 is None:
return True
if T1.val != T2.val:
return self.isSubtree(T1.left, T2) or self.isSubtree(T1.right, T2)
else:
if self.isSametree(T1, T2):
return True
else:
return self.isSubtree(T1.left, T2) or \
self.isSubtree(T1.right, T2)
def isSametree(self, T1, T2):
if (T1 is None) and (T2 is None):
return True
if T1 is None:
return False
if T2 is None:
return False
if T1.val == T2.val:
return self.isSametree(T1.left, T2.left) and \
self.isSametree(T1.right, T2.right)
else:
return False