1. 题目
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
2. 思路
注意是B是A的子结构,并不是子树
也就有可能是,A是一颗很大的树,B只是中间一部分树的结构
所以要做的事情就两件
- 遍历A树,将A树每个节点与B树比较
- 判断A是否是B的结构(根节点相同)
代码
var compare = (p1, p2) => {
// 如果p2遍历空了,就代表p2属于p1的子结构,只是p1可能层级更深
if (!p2) return true
// 如果p1空了,p2没空,说明肯定就不属于了
else if (!p1) return false
if (p1.val === p2.val) {
const left = compare(p1.left, p2.left)
const right = compare(p1.right, p2.right)
return left && right
} else {
return false
}
}
function HasSubtree(pRoot1, pRoot2) {
if (!pRoot1 || !pRoot2) return false
// 遍历pRoot1的每个节点,去与pRoot2这棵树作比较
return compare(pRoot1, pRoot2) ||
HasSubtree(pRoot1.left, pRoot2) ||
HasSubtree(pRoot1.right, pRoot2)
}