剑指offer(17)树的子结构

时间:2022-07-05 14:48:38

题目描述

输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)

题目分析

分析如何判断树B是不是树A的子结构,只需要两步。很容易看出来这是一个递归的过程。一般在树的求解方面都和递归有关。

Step1.在树A中找到和B的根结点的值一样的结点R;

Step2.判断树A中以R为根结点的子树是不是包含和树B一样的结点。

代码

/* function TreeNode(x) {
this.val = x;
this.left = null;
this.right = null;
} */
function HasSubtree(pRoot1, pRoot2) {
let res = false;
if (pRoot1 === null || pRoot2 === null) return false;
if (pRoot1.val === pRoot2.val) res = doesTree1HasTree2(pRoot1, pRoot2);
if (!res) res = HasSubtree(pRoot1.left, pRoot2);
if (!res) res = HasSubtree(pRoot1.right, pRoot2);
return res;
}
function doesTree1HasTree2(pRoot1, pRoot2) {
if (pRoot2 === null) return true;
if (pRoot1 === null) return false;
if (pRoot1.val !== pRoot2.val) return false;
return doesTree1HasTree2(pRoot1.left, pRoot2.left) && doesTree1HasTree2(pRoot1.right, pRoot2.right);
}