输入两颗二叉树A,B,判断B是不是A的子结构。(PS:我们约定空树不是任意一个树的子结构)。
1、思路
要查找树A中是否存在和树B结构一样的子树,我们可以分为两步:第一步在树A中找到和B的根结点的值一样的结点R,第二步再判断树A中以R为根节点的子树是不是包含和树B一样的结构。
这里使用递归的方法即可。
使用先序遍历遍历A树,在A树中找到和B树根节点值相同的结点。
然后使用先序遍历+递归来判断B树中的结点是不是在A树中。
!!!递归的结束条件:如果B树中的结点空了,就返回true。如果是A树中的结点为null,则返回false。
2、步骤
1.首先需要判断A,B的根节点是否一样。
2.如果不一样,判断A的左孩子和B的根节点是否一样,同理可判断A的右孩子和B的根节点是否一样。依次找下去
如果上述情况都不满足则说明不包含
1.如果找到了A中有值和B中的根节点相同,则比较左右子树是否相同。
2.如果B为空了,则说明包含
3.如果A为空了,则说明不包含
JS代码:
function TreeNode(x) {
this.val = x;//写一个函数传入跟节点的值
this.left = null;
this.right = null;
}
function HasSubtree(pRoot1, pRoot2)
{
var res = false;
if(pRoot1 && pRoot2) {//AB的跟节点都存在,不为空
if(pRoot1.val === pRoot2.val) {
res = DoesTree1HavaTree2(pRoot1, pRoot2);//把结果暂存
}
if(!res) {//为假,再换成A 的左子树,去寻找这个子结构或者A的右子树去寻找这个子结构
res = HasSubtree(pRoot1.left, pRoot2) || HasSubtree(pRoot1.right, pRoot2);
}
}
return res;//res为真就返回
}
function DoesTree1HavaTree2(pRoot1, pRoot2) {
if(!pRoot2) {
return true;//B的节点空了,说明A包含B,真
}
if(!pRoot1) {
return false;//A空了,B有多余的,A不包含B
}
if(pRoot1.val !== pRoot2.val) {
return false;//AB都不空,比较每一个节点的值
}
return DoesTree1HavaTree2(pRoot1.left, pRoot2.left) && DoesTree1HavaTree2(pRoot1.right, pRoot2.right);//并的结果
}
module.exports = {
HasSubtree : HasSubtree
};
C++:
class Solution {
public:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool result = false;//标志默认为假
if(pRoot1 != NULL && pRoot2 != NULL){//两个跟节点都不是空,
if(pRoot1->val == pRoot2->val){//跟节点相同
result = DoesTree1HasTree2(pRoot1, pRoot2);//递归遍历
}
if(!result){
result = HasSubtree(pRoot1->left, pRoot2);//若跟不相同,就在左子树或者右子树去寻找
}
if(!result){
result = HasSubtree(pRoot1->right, pRoot2);
}
}
return result;
}
private:
bool DoesTree1HasTree2(TreeNode* pRoot1, TreeNode* pRoot2){
if(pRoot2 == NULL){
return true;//B空了就是对的
}
if(pRoot1 == NULL){
return false;//A空了就是错的
}
if(pRoot1->val != pRoot2->val){
return false;//值不相等返回假
}
return DoesTree1HasTree2(pRoot1->left, pRoot2->left) && DoesTree1HasTree2(pRoot1->right, pRoot2->right);//A和B的递归左子树和右子树应该都是相等才是对的。
}
};