剑指 Offer 26. 树的子结构 | 算法

73 阅读1分钟

剑指 Offer 26. 树的子结构

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

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

示例 1:

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

示例 2:

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

限制:

0 <= 节点个数 <= 10000

递归方法

var isSubStructure = function(A, B) {
    //空树不是任何一个树的子结构
    if (!A || !B) {
        return false;
    }
    return isSameTree(A, B) || isSubStructure(A.left, B) || isSubStructure (A.right, B);
};

    var isSameTree = function(A, B) {
            //B为空的时候,说明B已经遍历到底部了,证明B树完全在A树之中。返回True
    if (!B) {   
        return true;
    }
    //A树为空,且不满足B为空,说明A树已经遍历完了还没符合上面的情况,证明B树不存在这条分支
    if (!A) {   
        return false;
    }
    if (A.val !== B.val) {
        return false;
    }

    //递归考察左右子树
    return isSameTree(A.left, B.left) && isSameTree(A.right, B.right);
    }