剑指 Offer 26. 树的子结构

/**
 * 剑指 Offer 26. 树的子结构
 * @author wsq
 * @date 2020/09/10
	输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
	B是A的子结构， 即 A中有出现和B相同的结构和节点值。

	例如:
	给定的树 A:
	     3
	    / \
	   4   5
	  / \
	 1   2
	
	给定的树 B：
	   4 
	  /
	 1
	返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。
	
	示例 1：
	输入：A = [1,2,3], B = [3,1]
	输出：false

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

	限制：
	0 <= 节点个数 <= 10000
	
	链接：https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof
 */
package notsubmit;

public class TreeSubStructure {
	/**
	 * 1.采用先序遍历A的每一个节点
	 * 2.检查该节点对应的子结构是否包含B
	 * @param A
	 * @param B
	 * @return
	 */
	public boolean isSubStructure(TreeNode A, TreeNode B) {
        if(A == null || B == null){
            return false;
        }
        return check(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }
	
    public boolean check(TreeNode nodeA, TreeNode nodeB){
        if(nodeB == null){
            return true;
        }
        if(nodeA == null || nodeA.val != nodeB.val){
            return false;
        }
        return check(nodeA.left, nodeB.left) && check(nodeA.right, nodeB.right);
    }
}