题目描述
输入两颗二叉树A,B,判断B是不是A的子结构。
解法:就是先找到相等的根节点,然后再子树遍历。所以要用到两个递归函数,根节点的递归函数,子树的递归函数。
注意子树的递归函数中应该先判断子树是否遍历完为空,则返回true,每次遍历前都应该判断根节点是否存在。这样就完美了。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
bool IsSubtree(TreeNode* p1, TreeNode* p2) {
if(!p2) return true;//要先判断子树是否为空。这是为什么呢?如果恰好主树遍历完,子树也遍历完,那么就会出错先
if(!p1) return false;//需要判定根节点是否为空,这是遍历前的做法
if(p1->val != p2->val) return false;
return IsSubtree(p1->left, p2->left) && IsSubtree(p1->right, p2->right); //遍历子树左右子节点
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool flag = false;
if(pRoot1 && pRoot2) {
if(pRoot1->val == pRoot2->val) { //寻找到相等的根节点,然后再子树遍历
flag = IsSubtree(pRoot1, pRoot2);
}
if(!flag) flag = HasSubtree(pRoot1->left, pRoot2); //没有找到相应的根节点,左子树遍历
if(!flag) flag = HasSubtree(pRoot1->right, pRoot2); //右子树遍历,找到相等的根节点
}
return flag;
}
};