题目描述:
输入两棵二叉树A、B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解题思路:
遍历A树,找到与B树根节点相同的结点,再将它们的左右子树比较,直到B树为空。则返回true。
class Solution {
public:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(pRoot1 == NULL || pRoot2 == NULL) //前序遍历的出口
return false;
//前序遍历主树pRoot1各结点,从根节点到左子树再到右子树
if(is_same(pRoot1, pRoot2))
return true;
else //判断当前结点的左子树是否为子树pRoot2,再判断右子树是否为子树pRoot2
return HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2);
}
//确定父结点后,开始同时扫描pRoot1和pRoot2的各结点是否相等
bool is_same(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(pRoot2 == NULL) //如果遍历到pRoot2的空结点,返回true
return true;
if(pRoot1 == NULL) //如果pRoot1为空,而pRoot2不为空,说明不同,返回false
return false;
if(pRoot1->val == pRoot2->val)
return is_same(pRoot1->left, pRoot2->left) && is_same(pRoot1->right, pRoot2->right);
else
return false; //不相同返回false
}
};