题目是这样得:输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
1.分析
(1)首先设置标志位result = false,因为一旦匹配成功res就设为true,
剩下的代码不会执行,如果匹配不成功,默认返回false
(2)递归思想,如果根节点相同则递归调用IsTree1HaveTree2(),如果根节点不相同,则判断tree1的左子树和tree2是否相同,再判断右子树和tree2是否相同
(3)注意NULL的条件,HasSubTree中,如果两棵树都不为空才进行判断,IsTree1HasTree2中,如果Tree2为空,则说明第二棵树遍历完了,即匹配成功,
tree1为空有两种情况:
- 如果tree1为空并且tree2不为空说明不匹配;
- 如果tree1为空,tree2为空,说明匹配。
2.代码实现
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
bool IsTree1HaveTree2(TreeNode* pRoot1,TreeNode* pRoot2)
{
//树2走完,且每个节点都是和树1的节点相同
if(pRoot2==NULL)
return true;
//如果Tree2还没有遍历完,Tree1却遍历完了,返回false
if(pRoot1==NULL)
return false;
//如果其中有一个点没有对应上,返回false
if(pRoot1->val != pRoot2->val)
return false;
//如果根节点对应的上,那么就分别去子节点里面匹配
return IsTree1HaveTree2(pRoot1->left,pRoot2->left)
&& IsTree1HaveTree2(pRoot1->right,pRoot2->right);
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool res=false;
//当Tree1和Tree2都不为空的时候,才进行比较,否则直接返回false
if(pRoot1 && pRoot2)
{
//如果两树的根节点值相同再递归判断左右子树的节点值是否相同
if(pRoot1->val == pRoot2->val)
//以这个根节点为起点判断是否包含Tree2
res=IsTree1HaveTree2(pRoot1,pRoot2);
//如果找不到,那么就再去pRoot1的左儿子当作起点,去判断时候包含Tree2
if(!res)
res=HasSubtree(pRoot1->left, pRoot2);
//如果还找不到,那么就再去pRoot2的右儿子当作起点,去判断时候包含Tree2
if(!res)
res=HasSubtree(pRoot1->right, pRoot2);
}
return res;
}