树的子结构

来源:互联网 发布:大数据信息化时代 编辑:程序博客网 时间:2024/05/18 21:08

题目描述

输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
思路:
//1.首先需要递归pRoot1树,找到与pRoot2根一样的节点,这需要一个遍历
//2.找到相同的根节点后,要判断是否子树,仍需要一个一个遍历对比
//树的遍历我们一般就用递归来做,那么根据分析需要两个递归函数如下:

/*struct TreeNode {int val;struct TreeNode *left;struct TreeNode *right;TreeNode(int x) :val(x), left(NULL), right(NULL) {}};*/class Solution {public:    bool isSubtree(TreeNode* r1, TreeNode* r2){         //此处为p2 == null 是匹配完成的条件         //最开始p2肯定不为NULL,这是在主程序HasSubtree中判断过的。         //递归中,如果p2为空了,则表示上一层的递归中的p2已经匹配完了         if(r2 == NULL)              return true;        if(!r1)            return false;        if(r1->val == r2->val) //判断两个树的节点是否相等            return isSubtree(r1->left,r2->left) && isSubtree(r1->right,r2->right);//如果节点相等,判断左右子树是否分别相等        else            return false;    }    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)    {        if(pRoot1 == NULL || pRoot2 == NULL)            return false;        return isSubtree(pRoot1,pRoot2) || HasSubtree(pRoot1->left,pRoot2) || HasSubtree(pRoot1->right,pRoot2);        //短路特性,判断pRoot2是否是pRoot1的子树        //判断pRoot2是否是pRoot1左子树的子树        //判断pRoot2是否是pRoot1右子树的子树    }};