参考链接
题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
思路分析
要查找树A中是否存在和树B结构一样的子树,可以分成两步:
1.第一步在树A中找到和B的根节点的值一样的结点R;这实际上就是树的遍历。可以用递归实现递归调用HasSubTree遍历二叉树A。如果发现某一结点的值和树B的头结点的值相同,则转向第2步判断两个结点为根的数是否存在父子关系
2.第二步再判断树A中以R为根结点的子树是不是包含和树B一样的结构。这个过程其实就是要要判断两棵树对应的节点数据是否相同。这个是一个递归的过程
首先我们先实现第2步的操作,这个操作其实就是递归判断两个树对应节是否相同,递归的终结是如果之前的节点均相同,最后子树为空时,而父树如果也是NULL,则说明两颗树完全一样,如果父树不是NULL,则子树是父树的一部分
代码实现
function HasSubtree(pRoot1, pRoot2) {
if (pRoot2 == null || pRoot1 == null) {
return false;
}
let result;
//如果当前A树的根节点与B树的根节点相同,直接从A树当前位置开始判断
if (pRoot1.val == pRoot2.val) {
result = DoesAHasB(pRoot1, pRoot2)
}
if (!result) {
return HasSubtree(pRoot1.left, pRoot2) || HasSubtree(pRoot1.right, pRoot2)
} else {
return true
}
}
function DoesAHasB(pRoot1, pRoot2) { //判断树A和树B对应的值是否相等
if (pRoot2 == null) { //树B为空但是树A不为空,则树B肯定为树A的子结构
return true;
} else if (pRoot1 == null) { //树A为空但是树B不为空,则不存在
return false;
}
if (pRoot1.val != pRoot2.val) {
return false;
} else {
return DoesAHasB(pRoot1.left, pRoot2.left) && DoesAHasB(pRoot1.right, pRoot2.right)
}
}