本题运用的递归解法
题目:
输入两棵二叉树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
实现代码:
/**
* 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}
*/
var isSubStructure = function (A, B) {
// 判断A或B是否为null 或 A或B是否遍历到了末尾
if(!A || !B){ // **first**
return false;
}
// 寻找A中与B的根结点对应的结点
return isSameTree(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
};
function isSameTree(rootA, rootB){
// 既然到了这,则说明了A与B都不为空(因为经过了**first**处的筛选),所以这里的判断是判断下一次的结点(递归后的)
if(!rootB){ // 判断子节点是否为空,即B是否已经遍历到了末尾
return true;
}
if(!rootA){ // 若A遍历到了末尾,还没返回true的话,则返回false
return false;
}
if(rootA.val !== rootB.val){
return false;
}
return isSameTree(rootA.left, rootB.left) && isSameTree(rootA.right, rootB.right);
}
分析:
因为要判断B是否为A的子结构,所以必须要先找到B的根节点在A中的位置(第一个函数的递归要实现的目的),才可以开始往下匹配判断是否为A的子结构,若没找到则返回false,若找到了则进入isSameTree的递归,直到遍历到B的末尾依旧没有返回false,则返回true;或者直到遍历到A的末尾,B还没遍历到末尾,则意味着B中有子结点不在A中,所以不是A的子结构,返回false。