树的子结构

来源:互联网 发布:手机wifi挂机赚钱软件 编辑:程序博客网 时间:2024/05/23 01:17

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

1.分析

(1)首先设置标志位result = false,因为一旦匹配成功res就设为true,剩下的代码不会执行,如果匹配不成功,默认返回false

(2)递归思想,如果根节点相同则递归调用IsTree1HaveTree2(),如果根节点不相同,则判断tree1的左子树和tree2是否相同,再判断右子树和tree2是否相同

(3)注意NULL的条件,HasSubTree中,如果两棵树都不为空才进行判断,IsTree1HasTree2中,如果Tree2为空,则说明第二棵树遍历完了,即匹配成功,

tree1为空有两种情况:
  • 如果tree1为空并且tree2不为空说明不匹配;
  • 如果tree1为空,tree2为空,说明匹配。

2.代码实现

struct TreeNode {int val;struct TreeNode *left;struct TreeNode *right;TreeNode(int x) :val(x), left(NULL), right(NULL) {}};
bool IsTree1HaveTree2(TreeNode* pRoot1,TreeNode* pRoot2)        {        //树2走完,且每个节点都是和树1的节点相同        if(pRoot2==NULL)            return true;        //如果Tree2还没有遍历完,Tree1却遍历完了,返回false        if(pRoot1==NULL)            return false;        //如果其中有一个点没有对应上,返回false        if(pRoot1->val != pRoot2->val)            return false;        //如果根节点对应的上,那么就分别去子节点里面匹配        return IsTree1HaveTree2(pRoot1->left,pRoot2->left)             && IsTree1HaveTree2(pRoot1->right,pRoot2->right);    }
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)    {        bool res=false;        //当Tree1和Tree2都不为空的时候,才进行比较,否则直接返回false        if(pRoot1 && pRoot2)         { //如果两树的根节点值相同再递归判断左右子树的节点值是否相同            if(pRoot1->val == pRoot2->val)//以这个根节点为起点判断是否包含Tree2                res=IsTree1HaveTree2(pRoot1,pRoot2);            //如果找不到,那么就再去pRoot1的左儿子当作起点,去判断时候包含Tree2            if(!res)                res=HasSubtree(pRoot1->left, pRoot2);           //如果还找不到,那么就再去pRoot2的右儿子当作起点,去判断时候包含Tree2            if(!res)                res=HasSubtree(pRoot1->right, pRoot2);        }            return res;    }