LeetCode第572题: 另一个树的子树

218 阅读1分钟

题干

给定两个非空二叉树 s 和 t,检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。

示例 1: 给定的树 s:

      3
     / \
    4   5
   / \
  1   2

给定的树 t:

    4 
   / \
  1   2

返回 true,因为 t 与 s 的一个子树拥有相同的结构和节点值。

来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/su… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法:递归

这道题的思想其实很简单,就是我们将s的每一个节点都与t树进行比较,看是否以s节点的根节点与t树相等即可。

如果到了叶子节点还不返回true我们就返回false。遍历整棵s树。

接着就是我们比较树与树是否相等的逻辑。首先看一下退出条件当我们root1和root2同时为叶子节点节点时说明我们返回true,当我们一个不为叶子节点时,说明我们结构不相同,结构不相同的两棵树的肯定也是不相同的。接着就是比较我们左右节点是否相等,还有当前节点的val是否相等。

代码实现

 /**
  * @param {TreeNode} root
  * @param {TreeNode} subRoot
  * @return {boolean}
  */
 var isSubtree = function (root, subRoot) {
     function doFindRoot(root, subRoot) {
         if (root == null) return false
         if (doFindsubRoot(root, subRoot)) {
             return true
         }
         let a = doFindRoot(root.left, subRoot)
         let b = doFindRoot(root.right, subRoot)
         return a || b
     }
     function doFindsubRoot(root1, root2) {
         if (root1 == null && root2 == null) return true
         if (root1 == null || root2 == null) return false
         let left=doFindsubRoot(root1.left, root2.left)
         let right=doFindsubRoot(root1.right, root2.right)
         if(left&&right&&root1.val == root2.val) return true
         
     }
     return doFindRoot(root, subRoot)
 };