题目:输入两个二叉树A和B,判断A是不是B的子结构。
如图下面这颗树就是上面的树的子结构。
思路:要查找树A中是否存在和树B结构一样的子树,我们可以分为两步:第一步在树A中找到和B的根节点的值一样的结点R,第二步在判断树A中以R为根节点的子树是不是包含和树B一样的结构。
第一步在树A中找与树B根节点的值一样的结点,这实际上就是树的遍历。
第二步是判断树A中以R为根节点的子树是不是和树B具有相同的结构。
递归来考虑:如果结点R的值和树B的根的值不一样,则以R为根节点的子树和树B肯定不具有相同的结点;如果他们的节点值相同,则递归判断他们各自的左右子树的结点的值是不是相同的。递归的终止条件是我们到达了树A或者树B的叶子节点。
bool DoTree1HaveTree2(BTNode* root1, BTNode* root2)
{
if (root1 == NULL)
{
return false;
}
if (root2 == NULL)
{
return true;
}
if (root1->_data != root2->_data)
{
return false;
}
return DoTree1HaveTree2(root1->_left, root2->_left) && DoTree1HaveTree2(root1->_right, root2->_right);
}
bool HasSubTree(BTNode* root1, BTNode* root2)
{
bool result = false;
if (root2 == NULL)
{
return true;
}
if (root1 == NULL)
{
return false;
}
if (root1->_data == root2->_data)
{
result = DoTree1HaveTree2(root1, root2);
}
if (!result)
{
result = HasSubTree(root1->_left, root2);
}
if (!result)
{
result = HasSubTree(root1->_right, root2);
}
return result;
}