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

Posted 2022-12-29 chenruibin0614

 

/*

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 || !pRoot2) return false;

        bool result=false;

        if(pRoot1->val == pRoot2->val)result=isSubtree(pRoot1,pRoot2); // 找到判断子树

        if(!result) result=HasSubtree(pRoot1->left,pRoot2); // 未找到匹配的根节点以及匹配的子树，则继续向下递归

        if(!result) result=HasSubtree(pRoot1->right,pRoot2);

        return result;

   

    

    bool isSubtree(TreeNode* pRoot1, TreeNode* pRoot2)

   

        if(!pRoot2) return true; // 子树遍历完成（关键语句）

        if(!pRoot1) return false; // 主树异常时的输出（关键语句：提高鲁棒性）

        bool result=true;

        if(pRoot1->val!=pRoot2->val) result=false;

        if(result) result=isSubtree(pRoot1->left,pRoot2->left);

        if(result) result=isSubtree(pRoot1->right,pRoot2->right);

        return result;

   

;

 

树的遍历问题使用递归事半功倍，这里是分别先序遍历两棵树，并比较节点值。需要先判断出匹配根节点的位置，再去遍历。