[LeetCode]（面试题26）树的子结构

题目

输入两棵二叉树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

解题思路

若树 B 是树 A 的子结构，则子结构树 B 的根节点可能为树 A 的任意一个节点。因此，判断树 B 是否是树 A 的子结构，需完成以下两步工作：
第一步，isSubStructure(A, B) ，在树 A 中查找与树 B 根节点的值一样的节点 a，即先序遍历树 A 中的每个节点寻找与树 B 根节点的值一样的节点 a：
特例处理：当 树 A 为空或树 B 为空 时，直接返回 false ；
返回值（若树 A 中存在与树 B 根节点的值一样的节点 a，则必满足以下三种情况之一，因此用 || 连接）：
1）树 A 的根节点与树 B 的根节点值一样，则继续判断树 A 与树 B 的剩余节点是否一样，对应 dfs(A, B)；
2）树 A 左子树中存在与树 B 根节点的值一样的节点，对应 isSubStructure(A.left, B)；
3）树 A 右子树中存在与树 B 根节点的值一样的节点，对应 isSubStructure(A.right, B)；

第二步，dfs(A, B) 函数，判断树 A 中以 a 为根节点的子树是否包含树 B，即树 A 中以 a 为根节点的子树的某部分与树 B 的每个节点的值都能对应：
终止条件：
当节点 B 为空：说明树 B 已匹配完成（越过叶子节点），因此返回 true ；
当节点 A 为空：此时树 B 还没有匹配完成，而树 A 已经越过叶子节点，即匹配失败，返回 false ；
返回值（若树 A 中以 a 为根节点的子树的某部分与树 B 的每个节点的值都能对应，则以下三个条件需要同时满足，因此用 && 连接）：
1）判断树 A 根节点的值与树 B 根节点的值是否相等，即 A.val == B.val；
2）判断树 A 和树 B 的左子节点是否相等，即 dfs((A.left, B.left) ；
3）判断树 A 和树 B 的右子节点是否相等，即 dfs((A.right, B.right) ；

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if(A == null || B == null){
            return false;
        }
        return dfs(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }

    private boolean dfs(TreeNode A, TreeNode B){
        if(B == null){
            return true;
        }
        if(A == null){
            return false;
        }
        return (A.val == B.val) && dfs(A.left, B.left) && dfs(A.right, B.right);
    }
}