【二叉树】树的子结构/判断一个节点是否在二叉树中

树的子结构

问题分析
输入两棵二叉树A和B，判断B是不是A的子结构

思路：
1、在树A中找到和B根节点值一样的节点R
2、再判断树A中以R节点为根节点的子树是不是包含和树B一样的结构。

代码实现

//步骤1，找一样的节点
bool HasSubTree(TreeNode* pRoot1, TreeNode * pRoot2)
{
    bool res = false;
    if (pRoot1 != NULL && pRoot2 != NULL)
    {
        if (pRoot1->val == pRoot2->val)
            res = DoesTree1HaveTree2(pRoot1, pRoot2);

        //以该节点为根节点的结构不是子树结构
        if (!res)
            res = HasSubTree(pRoot1->left, pRoot2);
        if (!res)
            res = HasSubTree(pRoot1->right, pRoot2);
    }
    return res;
}
//步骤2：在A中找到与B根节点一样的点R，判断是否有一样的结构
//递归的思路来考虑，如果两个根节点相等的话，则判断他们得左右子树是否相等，
//直到到达其中一个数的叶节点
bool DoesTree1HaveTree2(TreeNode* pRoot1, TreeNode * pRoot2)
{
    if (pRoot2 == NULL)//如果B结构到达叶子节点，说明匹配
        return true;
    if (pRoot1 == NULL)//如果A中到达叶子结点，但是B没有到达叶子节点，说明不匹配
        return false;
    if (pRoot1->val != pRoot2->val)
        return false;
    return DoesTree1HaveTree2(pRoot1->left, pRoot2->left)
        && DoesTree1HaveTree2(pRoot1->right, pRoot2->right);
}

变形，判断两个树的结构是否相等

分析：其实这个问题与上一个问题类似，但是相对容易一些。
只是判断两个数的结构是否相等，不考虑到树的节点内容。

bool StructureCmp(Node *pRoot1,Node *pRoot2)
{
    //两个根节点都为空，返回真
    if(pRoot1 == NULL && pRoot2 == NULL)
        return true;
    if(pRoot1 == NULL || pRoot2 == NULL)
        return false;
    bool left = StructureCmp(pRoot1->left,pRoot2->left);
    bool right = StructureCmp(pRoot2->right,pRoot2->right);
    return (left&&right);
}

判断一个节点是否在二叉树中

思路：递归的方法，判断根节，然后 找左子树，找右子树

bool IsInTree(Node* pRoot,Node* find)
{
    if(pRoot == NULL || find == NULL)
        return false;
    if(pRoot == find)
        return true;
    //不是根节点，找左找右
    if(IsInTree(pRoot->left,find))
        return true;
    return IsInTree(pRoot->right,find);
}