Python3解leetcode Subtree of Another Tree

问题描述：

Given two non-empty binary trees s and t, check whether tree t has exactly the same structure and node values with a subtree of s. A subtree of s is a tree consists of a node in s and all of this node's descendants. The tree s could also be considered as a subtree of itself.

Example 1:
Given tree s:

Given tree t:

Return true, because t has the same structure and node values with a subtree of s.

Example 2:
Given tree s:

Given tree t:

Return false.

解题思路：

关于树的题目，第一反应就是利用DFS解答，此题也不例外。

代码：

 class Solution:

     def isSubtree(self, s: TreeNode, t: TreeNode) -> bool:

         def dfs(a,b):

             if not a or not b:  #若果a,b中存在null，处理手段

                 return not a and not b

             #以下处理时在a,b皆不为null的情况下进行讨论

             if a.val==b.val and dfs(a.left,b.left) and dfs(a.right,b.right):

                 return True

             if b is t:#当b时t的时候，判断a的左右子树分别与t是否相等

                 return dfs(a.left,t) or dfs(a.right,t)

             return False

         return dfs(s,t)

第4、5行代码，将各种子树为空的情形缩略到一个条件判断中，精简了代码。

第7、8行代码，判断当前树是否和t树完全相同

第9、10行代码，只有当b是t的时候才生效，意思是如果该次执行是从第二个if递归过来的，那么就不进行判断，因为该次执行判断是当前树的子树与t的子树是否相同，并非是判断t是否与当前树相同。

最后12行代码，直接返回False即可。但如果返回的是dfs(a,t)，代码执行时间会缩短75%，但是我没懂为何会缩短如此之多。

巴特西

Python3解leetcode Subtree of Another Tree

最新文章

热门文章