JZ17 树的子结构**
题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
思路
两步走:
(1)在树A中找到和树B的根节点的值一样的节点R
(2)判断树A中以R为根节点的子树是不是包含和树B一样的结构
第一步在树A中查找与B根节点一样的节点,需要对树进行遍历(递归)
注意:一定要注意空指针
当树A或树B为空的时候,定义相应的输出。如果没有进行检查并进行相应的处理,程序会非常容易崩溃!!!
代码如下:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2) { bool result=false; //只有当树1和树2都不为空时才有继续的必要,否则直接返回false if(pRoot1!=NULL&&pRoot2!=NULL) { if(pRoot1->val==pRoot2->val) //找到树1中节点与树2中节点相同的 { result=DoesTree1HaveTree2(pRoot1,pRoot2); //接下去判断相同节点下的这棵树的结构是否相同 } //如果找不到,就继续找树1的左子树 if(!result) result=HasSubtree(pRoot1->left,pRoot2); //如果还是找不到,就继续找树1的右子树 if(!result) result=HasSubtree(pRoot1->right,pRoot2); } return result; }
上述代码中递归调用HasSubtree遍历二叉树A,如果发现某一节点的值和树B的头结点的值相同,则调用DoesTree1HaveTree2,进行第二步判断
第二步:判断树A中以R为根节点的子树是不是和树B具有相同的结构。同样可以用递归的思路来考虑:如果节点R的值和树B的根节点不同,则以R为根节点的子树和树B肯定不具有相同的节点;如果它们的值相同,则递归地判断它们各自的左右节点的值是否相同。递归的终止条件时到达了树A或树B 的叶子节点。
代码如下:
bool DoesTree1HaveTree2(TreeNode* pRoot1,TreeNode*pRoot2) { //下面四行是递归的终止条件:到达了树A或树B的叶子节点 //树2遍历结束,返回true if(pRoot2==NULL) return true; //树2遍历完了,树1还没有遍历完,返回false if(pRoot1==NULL) return false; //如果某一个节点不一样,则返回false if(pRoot1->val!=pRoot2->val) { return false; } //分别判断两棵树的左右节点是否相等 return DoesTree1HaveTree2(pRoot1->left,pRoot2->left)&&DoesTree1HaveTree2(pRoot1->right, pRoot2->right); }
代码
class Solution { public: bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2) { bool result=false; //只有当树1和树2都不为空时才有继续的必要,否则直接返回false if(pRoot1!=NULL&&pRoot2!=NULL) { if(pRoot1->val==pRoot2->val) //找到树1中节点与树2中节点相同的 { result=DoesTree1HaveTree2(pRoot1,pRoot2); //接下去判断相同节点下的这棵树的结构是否相同 } //如果找不到,就继续找树1的左子树 if(!result) result=HasSubtree(pRoot1->left,pRoot2); //如果还是找不到,就继续找树1的右子树 if(!result) result=HasSubtree(pRoot1->right,pRoot2); } return result; } //判断两棵树的结构是否一样 bool DoesTree1HaveTree2(TreeNode* pRoot1,TreeNode*pRoot2) { //树2遍历结束,返回true if(pRoot2==NULL) return true; //树2遍历完了,树1还没有遍历完,返回false if(pRoot1==NULL) return false; //如果某一个节点不一样,则返回false if(pRoot1->val!=pRoot2->val) { return false; } //分别判断两棵树的左右节点是否相等 return DoesTree1HaveTree2(pRoot1->left,pRoot2->left)&&DoesTree1HaveTree2(pRoot1->right, pRoot2->right); } };