suibenzhi
7/13/2018 - 3:45 PM

树的子结构.cpp

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


bool doesTree1HaveTree2(TreeNode *node1, TreeNode *node2) {
    //如果Tree2已经遍历完了都能对应的上,返回true
    if (node2 == NULL) {
        return true;
    }
    //如果Tree2还没有遍历完,Tree1却遍历完了。返回false
    if (node1 == NULL) {
        return false;
    }
    //如果其中有一个点没有对应上,返回false
    if (node1->val != node2->val) { 
            return false;
    }
      
    //如果根节点对应的上,那么就分别去子节点里面匹配
    return doesTree1HaveTree2(node1->left, node2->left) && doesTree1HaveTree2(node1->right, node2->right);
}
 
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
    bool result = false;
    //当Tree1和Tree2都不为零的时候,才进行比较。否则直接返回false
    if (pRoot1 != NULL && pRoot2 != NULL) {
        //如果找到了对应Tree2的根节点的点
        if(pRoot1->val == pRoot2->val){
            //以这个根节点为为起点判断是否包含Tree2
            result = doesTree1HaveTree2(pRoot1,pRoot2);
        }
        //如果找不到,那么就再去root的左儿子当作起点,去判断时候包含Tree2
        if (!result) {
            result = HasSubtree(pRoot1->left,pRoot2);
        }
          
        //如果还找不到,那么就再去root的右儿子当作起点,去判断时候包含Tree2
        if (!result) {
            result = HasSubtree(pRoot1->right,pRoot2);
        }
    }
        //返回结果
    return result;
}