输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)

 

/*

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;

    }

};

 

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

posted @ 2019-10-06 16:12  玄虚子  阅读(430)  评论(0)    收藏  举报