力扣 剑指 Offer 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的子结构即可判断为true，不需要到叶子节点。

解题步骤分为两步：

1、先将根节点匹配

2、根节点匹配后，对子树进行匹配。

/**
 * 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;   // B全匹配到了，返回true
        if(A == null || A.val != B.val) return false;  // A没了或者值不相等，返回false
        return dfs(A.left, B.left) && dfs(A.right, B.right);  //进入左孩子和右孩子进行比较
    }
}