[路飞] 39——leetcode – 剑指 Offer 26. 树的子结构

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

剑指 Offer 26. 树的子结构

题目分析

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

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

例如:
给定的树 A:

     3
    / \
   4   5
  / \
 1   2
给定的树 B：

   4 
  /
 1
返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。
复制代码

思路讲解

我们需要判断A树是否包含B树

• ⾸先我们就需要现在A树中找到B树的根节点
• 找到B树的根节点后，再去判断B树根节点下⾯的⼦节点是否和A树中找到节点的⼦节点相等
• 直到整个B树都在A树中找到，或者有节点不同

示例

输入： A = [1,2,3], B = [3,1]
输出： false

输入： A = [3,4,5,1,2], B = [4,1]
输出： true
复制代码

代码

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} A
 * @param {TreeNode} B
 * @return {boolean}
 */
 /**
 * 我们先在A树中寻找B树的根节点，如果没找到，就返回false
 * 如果找到了，我们就判断A树中找到的节点的左右子树，是否和B树根节点的左右子树相同
 * 如何判断是否相同，参照前面两步（递归）
  */
var isSubStructure = function(A, B) {
    return (!!A && !!B) && (recur(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B))
};
var recur = function (A, B) {
    if (!B) return true
    if (!A || A.val !== B.val) return false
    return recur(A.left, B.left) && recur(A.right, B.right)
}
复制代码