剑指 Offer 26. 树的子结构

剑指 Offer 26. 树的子结构

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

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

例如:
给定的树 A:

   3
  / \
 4   5
  / \
 1   2
给定的树 B：

    4 
  /
 1

返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

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

两次遍历，第一次遍历先遍历出开始的节点，第二次遍历开始比较。

先从根节点判断B是不是A的子结构，如果不是在分别从左右两个子树判断。只有有一个节点符合条件，结果就是正确的。

class Solution {
    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if (A == null || B == null)
            return false;
        return checkSub(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }

    boolean checkSub(TreeNode A, TreeNode B) {
        //B已经访问完了，确定是A的子结构
        if (B == null)
            return true;
        //如果B不为空A为空，或者这两个节点值不同，说明B树不是A的子结构，直接返回false
        if (A == null || A.val != B.val)
            return false;
        return checkSub(A.left, B.left) && checkSub(A.right, B.right);
    }
}