[剑指 Offer 26. 树的子结构]

50 阅读1分钟

剑指 Offer 26. 树的子结构

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

B是A的子结构, 即 A中有出现和B相同的结构和节点值。

image.png

示例 1:

输入: A = [1,2,3], B = [3,1]
输出: false

示例 2:

输入: A = [3,4,5,1,2], B = [4,1]
输出: true

思路:先在A中找到于B根节点相同的节点,再进行比较判断

//判断函数,利用递归实现比较
bool  Compare(TreeNode* A, TreeNode* B){
    // 若B已经无后续,则表明递归完毕,BA的子结构
    if(B == NULL)   return true;
    // 若B不为空并且A为空或者AB的值不相等,直接可以判断B不是A的子结构
    if(A == NULL || A->val != B->val)     return false;
    //该节点相同,则进入下一层比较
    return Compare(A->left,B->left)&&Compare(A->right,B->right);
}

这里有个大错误:当A->val==B->val时,直接return 就寄了

//在A中找B节点相同的节点(错误版本)
bool isSubStructure(TreeNode* A, TreeNode* B) {

    if(A==NULL||B==NULL) return false;
    //如果A B指向的节点相同,则进入判断是否具有相同的结构的函数
    if(A->val==B->val) return Compare(A,B);
    return isSubStructure(A->left,B)||isSubStructure(A->right,B);
}

image.png

//在A中找B节点相同的节点
bool isSubStructure(TreeNode* A, TreeNode* B) {
    if(A==NULL||B==NULL) return false;
    //有可能后面还有相同的节点入口,所以不能直接return
    return Compare(A,B)||isSubStructure(A->left,B)||isSubStructure(A->right,B);
}