树的子结构

Posted 2020-11-14 lune-qiu

题目描述

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

 

思路：看到这个题我们基本的思路就是，判断树A和树B的每一个节点，找到相同的节点然后就继续比较。

那么就是递归的思想。

代码：

   bool DFS(TreeNode* pRoot1,TreeNode* pRoot2)
    {
        if(!pRoot2)
            return true;
        if(!pRoot1)
            return false;
        if(pRoot1->val != pRoot2->val)
            return false;
        return DFS(pRoot1->left , pRoot2->left) && DFS(pRoot1->right , pRoot2->right);
    }
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot1 == NULL || pRoot2 == NULL)
            return false;
        return DFS(pRoot1,pRoot2)||
            HasSubtree(pRoot1->left,pRoot2)|| 
            HasSubtree(pRoot1->right,pRoot2);
    }

 

函数HasSubTree：将pRoot1本身，左孩子，右孩子，分别于pRoot2递归的比较 ，直到平pRoot1的所有节点都遍历完。pRoot2不能为NULL的原因是空树不是任何树的子树。

函数DFS:如果找到了相同的节点则继续递归的比较下去，如果两个节点的值不同，或者pRoot1遍历到结尾但pRoot2没有，则返回flase,如果pRoot2为NULL则说明pRoot1包含了pRoot2.