【LeetCode-面试题】面试题26. 树的子结构

面试题26. 树的子结构

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

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

解题思路：dfs

注意子结构和子树的区别

• 子树: 整个树要与原树相同, A与B相同, A的孩子与B相同
• 子结构: 只有有部分相同即可, 当B遍历到NULL时, 即返回true

/**
 * 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都走完了还没找到，那应该就是找不到了
        if(A == null || B == null) {
            return false;
        }
        return check(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }
    
    public boolean check(TreeNode A, TreeNode B) {
        // B遍历完了，就可以判断B是A的子结构
        if(B == null) {
            return true;
        }
        // 当节点 A 为空：说明已经越过树 A 叶子节点，即匹配失败
        if(A == null) {
            return false;
        }
        
        if(A.val != B.val) {
            return false;
        }
        
        return check(A.left, B.left) && check(A.right, B.right);
    }
}