剑指 Offer 26. 树的子结构

94 阅读1分钟

剑指 Offer 26. 树的子结构

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

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

示例 1:

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

示例 2:

输入: A = [3,4,5,1,2], B = [4,1]
输出: true

在leecode上使用数组去简单表示树的结构,其实我们自己可以模拟树的结构 二叉树模拟对象

var tree = {
    val:1,
    left:{
        val:2
        left:{...},
        right:{...}
    },
    right:{
        val:3
        left:{...},
        right:{...}        
    }
}

下面是代码实现

// 输入两棵二叉树A和B,判断B是不是A的子结构,(约定空树不是任意一个树的子结构)
var isSubStructure = function (A, B) {
  if (!A || !B) return false
  return isSame(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B)
  function isSame(A, B) {
    if (!B) return true // A和B都是一个叶节点,他们的左右子节点做比较,会走这里判断
    if (!A) return false // B存在,A不存在
    if (A.val !== B.val) return false
    return isSame(A.left, B.left) && isSame(A.right, B.right)
  }
}