题目:输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
要查找树A中是否存在和树B结构一样的子树,可以分成两步:
1.第一步在树A中找到和B的根节点的值一样的结点R;
这实际上就是树的遍历。可以用递归实现
递归调用HasSubTree遍历二叉树A。如果发现某一结点的值和树B的头结点的值相同,则转向第2步判断两个结点为根的数是否存在父子关系
2.第二步再判断树A中以R为根结点的子树是不是包含和树B一样的结构。
这个过程其实就是要要判断两棵树对应的节点数据是否相同。这个是一个递归的过程。
首先我们先实现第2步的操作,这个操作其实就是递归判断两个树对应节是否相同,
递归的终结是如果之前的节点均相同,最后子树为空时,而父树如果也是NULL,则说明两颗树完全一样,如果父树不是NULL,则子树是父树的一部分
牛客网上的一个大佬:
下面是对上述递归的一个解释代码:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
bool isSubtree(TreeNode* pRootA, TreeNode* pRootB)
{
if(!pRootB)return true;
if(!pRootA)return false;
if(pRootA->val==pRootB->val)
{
if(isSubtree(pRootA->left,pRootB->left)&&isSubtree(pRootA->right,pRootB->right))
return true;
}
return false;
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(!pRoot2||!pRoot1){return false;}
if(pRoot1->val==pRoot2->val)
{
if(isSubtree(pRoot1,pRoot2))
return true;
}
if(HasSubtree(pRoot1->left,pRoot2))
{
return true;
}
if(HasSubtree(pRoot1->right,pRoot2))
{
return true;
}
return false;
}
};