剑指offer-树的子结构

题目描述

 

输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

 

解题思路

考虑构造HasSubTree函数前序遍历树A递归判断子结构。从树A的根节点开始每遍历到一个节点，判断其结点值是否与树B的根节点值相等，若相等则调用IsSub函数判断A的此子树是否包含树B。具体来说，IsSub函数从树B的根节点开始与子树根节点比较，若不等则直接返回false，否则递归比较树B的左子树和右子树直到叶子节点，如果遍历完树B发现结点值都相等，则返回true。回到HasSubTree函数中，若此子树包含树B，那么直接返回true，否则继续递归判断左子树与右子树是否包含树B。若遍历完树A仍未发现树B，就返回false。

代码

/*
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)
    {
        if(pRoot1==NULL||pRoot2==NULL)
            return false;
        if(pRoot1->val==pRoot2->val)
            if(IsSub(pRoot1,pRoot2))
                return true;
        else{
            if(pRoot1->left)
                if(HasSubtree(pRoot1->left,pRoot2))
                    return true;
            if(pRoot1->right)
                if(HasSubtree(pRoot1->right,pRoot2))
                    return true;
        }
        return false;
    }
    bool IsSub(TreeNode* pRoot1, TreeNode* pRoot2){
        if(pRoot1->val!=pRoot2->val)
            return false;
        if(pRoot2->left){
            if(pRoot1->left){
                if(!IsSub(pRoot1->left,pRoot2->left))
                    return false;
            }
            else 
                return false;
        }
        if(pRoot2->right){
            if(pRoot1->right){
                if(!IsSub(pRoot1->right,pRoot2->right))
                    return false;
            }
            else
                return false;
        }
        return true;
    }
};