题意:输入两棵二叉树A 和B,判断B 是不是A 的子结构。二叉树结点的定义如下:
struct BinaryTreeNode{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
};
要查找树A 中是否存在和树B 结构一样的子树,我们可以分成两步: 第一步在树A 中找到和B 的根结点的值一样的结点R, 第二步再判断树A 中以R 为根结点的子树是不是包含和树B 一样的结构。
第一步在树A 中查找与根结点的值一样的结点, 这实际上就是树的遍历。对二叉树这种数据结构熟悉的读者自然知道可以用递归的方法去遍历,也可以用循环的方法去遍历。由于递归的代码实现比较简沽,面试时如果没有特别要求,我们通常都会采用递归的方式。
代码如下:
bool HasSubtree(BinaryTreeNode* pRoot1,BinaryTreeNode*pRoot2){
bool result=false;
if(pRoot1!=NULL&&pRoot2!=NULL){
if(pRoot1->m_nValue==pRoot2->m_nValue)//第一步
result=DoesTree1HaveTree2(pRoot1,pRoot2);//第二步
if(!result)
result=HasSubtree(pRoot1->m_pLeft,pRoot2);
if(!result)
result=HasSubtree(pRoot1->m_pRight,pRoot2);
}
return result;
}
bool DoesTree1HaveTree2(BinaryTreeNode* pRoot1,BinaryTreeNode* pRoot2){
if(pRoot2==NULL)//说明A的子树遍历完成,子树相等
return true;
if(pRoot1==NULL)//说明A的子树遍历没完成,但B的子树已经遍历完了,说明子树不相等
return false;
if(pRoot1->m_nValue!=pRoot2->m_nValue)
return false;
return DoesTree1HaveTree2(pRoot1->m_pLeft,pRoot2->m_pLeft)&&DoesTree1HaveTree2(pRoot1->m_pRight,pRoot2->m_pRight);
}
在上述代码中,我们递归调用HasSubtree 遍历二叉树A(是为了发现与树B的头结点的值相等的结点)。如果发现某一结点的值和树B 的头结点的值相同,则调用DoesTree1HaveTree2 ,做第二步判断。
第二步是判断树A 中以R 为根结点的子树是不是和树B 具有相同的结构。同样,我们也可以用递归的思路来考虑:如果结点R 的值和树B 的根结点不相同,则以R 为根结点的子树和树B 肯定不具有相同的结点:如果它们的值相同,则递归地判断它们各自的左右结点的值是不是相同。递归的终止条件是我们到达了树A 或者树B 的叶结点。