|
| 1 | +# Definition for a binary tree node. |
| 2 | +# class TreeNode(object): |
| 3 | +# def __init__(self, val=0, left=None, right=None): |
| 4 | +# self.val = val |
| 5 | +# self.left = left |
| 6 | +# self.right = right |
| 7 | + |
| 8 | +# My initial approacch |
| 9 | +class Solution(object): |
| 10 | + def isSubtree(self, s, t): |
| 11 | + """ |
| 12 | + :type s: TreeNode |
| 13 | + :type t: TreeNode |
| 14 | + :rtype: bool |
| 15 | + """ |
| 16 | + startingNodesOnS = [] |
| 17 | + self.getStartingNodeOnS(s, t.val, startingNodesOnS) |
| 18 | + if len(startingNodesOnS) <= 0: |
| 19 | + return False |
| 20 | + |
| 21 | + for node in startingNodesOnS: |
| 22 | + if self.isSameTree(node, t): |
| 23 | + return True |
| 24 | + return False |
| 25 | + |
| 26 | + def getStartingNodeOnS(self, s, tVal, startingNodesOnS): |
| 27 | + if not s: |
| 28 | + return |
| 29 | + if s.val == tVal: |
| 30 | + startingNodesOnS.append(s) |
| 31 | + leftNode = self.getStartingNodeOnS(s.left, tVal, startingNodesOnS) |
| 32 | + rightNode = self.getStartingNodeOnS(s.right, tVal, startingNodesOnS) |
| 33 | + if leftNode: |
| 34 | + startingNodesOnS.append(leftNode) |
| 35 | + return |
| 36 | + elif rightNode: |
| 37 | + startingNodesOnS.append(rightNode) |
| 38 | + return |
| 39 | + else: |
| 40 | + return |
| 41 | + |
| 42 | + def isSameTree(self, startingNodeOnS, t): |
| 43 | + if not startingNodeOnS and not t: |
| 44 | + return True |
| 45 | + if startingNodeOnS and t and startingNodeOnS.val == t.val: |
| 46 | + left = self.isSameTree(startingNodeOnS.left, t.left) |
| 47 | + right = self.isSameTree(startingNodeOnS.right, t.right) |
| 48 | + return left and right |
| 49 | + else: |
| 50 | + return False |
| 51 | + |
| 52 | + |
0 commit comments