题目描述
输入两颗二叉树A,B,判断B是不是A的子结构。
解题思路
1.首先需要递归pRoot1树,找到与pRoot2根一样的节点,这需要一个遍历
2.找到相同的根节点后,要判断是否子树,仍需要一个一个遍历对比
树的遍历我们一般就用递归来做,那么根据分析需要两个递归函数。
其中需要注意的是:
1. 测试用例如果pRoot2为空的话,返回的false而不是我们认为的空树应该是所有树的子树
2. 再判断是否子树的过程中,应该先判断pRoot2是否为空,为空则表明子树的所有节点都比较完了,应该是子树返回True
3. 要养成一个习惯,对任何一个树节点进行访问时,一定要提前检测该节点是否为空
代码
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool result = false;
if(pRoot1 != NULL && pRoot2 != NULL)
{
if (pRoot1->val == pRoot2->val){
result = DoesTree1HaveTree2(pRoot1,pRoot2);
}
if (!result){
result = HasSubtree(pRoot1->left,pRoot2);
}
if (!result){
result = HasSubtree(pRoot1->right,pRoot2);
}
}
return result;
}
bool DoesTree1HaveTree2(TreeNode* pRoot1,TreeNode* pRoot2){
if (pRoot2 == NULL){
return true;
}
if (pRoot1 == NULL){
return false;
}
if (pRoot1->val != pRoot2->val){
return false;
}
return DoesTree1HaveTree2(pRoot1->left,pRoot2->left) &&
DoesTree1HaveTree2(pRoot1->right,pRoot2->right);
}
};