树的子结构

题目要求

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构， 即 A中有出现和B相同的结构和节点值。

分析

将以树A的每一个节点的树和B相比较是不是结构相同

/**
 * 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) {
        // A 为空时 匹配失败，由于孔数不作为任何一个数的子结构，所以B为null时返回false
        if (A == null || B == null) {
            return false;
        }
        // 判断以A为头的树中是不是包含B || 判断以A的右子树为头的树中是否包含B || 判断以A的左子树为头的树中是否包含B
        return isSame(A, B) || isSubStructure(A.right, B) || isSubStructure(A.left, B);
    }

    // 判断以A为头的树中是否包含B
    public boolean isSame(TreeNode A, TreeNode B) {
        // B 为空表示已经匹配完, 表明A中包含B
        if (B == null) {
            return true;
        }
        // A已经空了，但是B此时还不空，表明没有匹配成功
        if (A == null) {
            return false;
        }

        // 判断 A的val是否与B的val是否相同 || A的左子树和B的左子树是否相同 || A的右子树与B的右子树是否相同
        return A.val == B.val && isSame(A.left, B.left) && isSame(A.right, B.right);
    }
}