剑指 Offer 26. 树的子结构
难度中等590
输入两棵二叉树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
思路
看下面的注解,我已经写的很详细了,主要难点,我觉得就是,在脑子里,构建递归图
题目算法代码如下
public class 树的子结构 {
public boolean isSubStructure(TreeNode A, TreeNode B) {
//只有一个树为空,直接返回true
if (A==null||B==null){
return false;
}
//这里有三种情况
//第一种 isSub(A,B) : 他们的头节点就开始匹配 (新建方法递归)
//第二种 isSubStructure(A.left,B) 子树在母树的左边 (原来方法递归)
//第二种 isSubStructure(A.left,B) 子树在母树的右边 (原来方法递归)
return isSub(A,B)||isSubStructure(A.left,B)||isSubStructure(A.right,B);
}
//能进入的这个方法,判断的是两棵树的 头节点齐头并进
private boolean isSub(TreeNode A, TreeNode B) {
//B ==null ,说明之前已经匹配成功发,没有返回false 那就是true
if (B==null){
return true;
}
//如果 A==null, 能从上面下来说明B不为空 或者不相等(齐头并进),肯定false
if (A==null||A.val!=B.val){
return false;
}
//同时遍历左子树,同时右子树
return isSub(A.left,B.left)&&isSub(A.right,B.right);
}
}