树的子结构

剑指Offer_18 树的子结构

---

2018/5/21 星期一

题目：输入两颗二叉树A和B，判断B是不是A的子结构。二叉树的定义如下：

class BinaryTreeNode {
    int data;
    BinaryTreeNode leftNode;
    BinaryTreeNode rightNode;
}

例如：下图中的两棵树，由于A中有一部分子树的结构和B是一样的，因此B是A的子结构。

两个二叉树A和B

分析：要查找树A中是否存在和树B一样的子结构，我们可以分为两步：第一步在树A中找到和树B的根结点值一样的结点R，第二步，再判断A中以R为根结点的子树是不是包含和树B一样的结构。

第一步

在A结点中查找根结点的值一样的结点，这实际上就是树的遍历，既可以用递归也可以用循环。通常递归比较简洁，下面使用递归方法来实现。参考代码：

boolean HasSubTree(BinaryTreeNode pRoot1, BinaryTreeNode pRoot2) {
    boolean result = false;
    // 遍历
    if (pRoot1 != null && pRoot2 != null) {
        if (pRoot1.data == pRoot2.data) {
            result = DoesTree1HaveTree2(pRoot1, pRoot2);
        }
        if (!result) {
            result = HasSubTree(pRoot1.leftNode, pRoot2);
        }
        if (!result) {
            result = HasSubTree(pRoot1.rightNode, pRoot2);
        }
    }
    return result;
}

在编写程序的时候，一定要注意边界条件的检查，也就是空指针。上面的代码中，如果发现第一结点的值相同，则调用DoesTree1HaveTree2，做第二步的判断。

第二步

判断树A和树B中以R为根结点的子树是不是和树B具有相同的结构。同样可以利用递归的思路来考虑：

1. 判断A和B结点的值，如果它们的值不相同，则肯定不是；
2. 如果相等，则开始判断左、右子树是否相等；
3. 递归的执行，执行的终止条件就是到达了树A或者树B的叶子结点。

参考代码如下：

boolean DoesTree1HaveTree2(BinaryTreeNode pRoot1, BinaryTreeNode pRoot2) {
    if (pRoot2 == null) {
        return true;
    }
    if (pRoot1 == null) {
        return false;
    }
    if (pRoot1.data != pRoot2.data) {
        return false;
    }
    return DoesTree1HaveTree2(pRoot1.leftNode, pRoot2.leftNode)
            && DoesTree1HaveTree2(pRoot1.rightNode, pRoot2.rightNode);
}

注意：二叉树相关的代码中，有大量的指针操作，每一次使用指针，我们都需要考虑，这个指针是否有可能为NULL。

测试用例：

1. 功能测试：树A和树B都是普通的二叉树；树B是或者不是树A的子树
2. 特殊用例测试：两个二叉树有一个或者两个为Null的根结点；二叉树的所有结点都没有左子树或者右子树。