【剑指offer】树的子结构

树的子结构

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

代码

递归

• 如果结点值相同：

  1. 如果这是B的根节点，递归检查r1,r2左孩子以及r1,r2右孩子，而且这两边递归都要相同才能说明B是A的子结构；
  2. 如果这不是B的根节点，递归检查r1左孩子和r2，以及递归检查r1右孩子和r2，两边任意一边相同都可以说明B是A的子结构；
  • 那么fun(r1,r2)=(fun(r1->left,r2->left) && fun(r1->right,r2->right) ) || fun(r1->left,r2) || fun(r1->right,r2)

• r2==NULL时，那么说明B的某个分支匹配正确，返回true

• 如果r1==NULL并且r2!=NULL,那么说明匹配失败，返回false

class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot1==NULL)
            return false;
        if(pRoot2==NULL)
            return false;
        return HS(pRoot1,pRoot2);
    }
    bool HS(TreeNode* pRoot1, TreeNode* pRoot2){
        if(pRoot2==NULL)
            return true;
        if(pRoot1==NULL)
            return false;
        if(pRoot1->val==pRoot2->val)
            return (HS(pRoot1->left,pRoot2->left) && HS(pRoot1->right,pRoot2->right))||HS(pRoot1->left,pRoot2) || HS(pRoot1->right,pRoot2);
        else
            return false;
    }
};