剑指Offer26.树的子结构
题目描述
输入两棵二叉树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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路和代码
思路过程
就是说啊,一拿到这样一个题目开始时有点懵的,开始想法时如果B是A的子结构那它的先序遍历字符串一定是A的字串,但是后面一想这个不是不一定,因为可能出现左右子节点一边为空时,先序遍历的结果时一致的。后面又仔细想了想其他方法,想到由于二叉树的特性,往往会用到一些诸如递归一样的方法,所以我就往递归方向想,即递归的去遍历A的所有节点,并判断B是否是以该节点为根节点的子树。然后在判断的时候又可以使用到递归,即从两树的根节点开始判断两个子树B是否是A的子结构。
代码及注释
class Solution {
//判断二叉树B是不是二叉树A的子结构
public boolean isSubStructure(TreeNode A, TreeNode B) {
//当A或B为null时,返回false
if(A == null || B == null){
return false;
}
//若A和B不为空,首先调用isSub方法,判断二叉树B是否为根节点与二叉树A根节点相同的子结构
//接着判断二叉树B是不是A的左子树或者右子树的子结构
return isSub(A,B) || isSubStructure(A.left,B) || isSubStructure(A.right,B);
}
//判断二叉树B是否为根节点与二叉树A根节点相同的子结构
//注意这个方法不是判断AB相同,而是判断从根节点(!!!)开始B是否是A的子结构
boolean isSub(TreeNode A,TreeNode B){
//如果B为空的话,就是B已经遍历完了,即B肯定是A的子结构,且A>>B的
if(B == null){
return true;
}
//如果A==null,则A先遍历完,即A<<B,B肯定不是A的子树
//如果当前节点A的值不等于B的值,那说明这两个节点不一样,B肯定不是A的子结构
if(A == null || A.val != B.val){
return false;
}
//递归的去判断AB的子树是否相同
return isSub(A.left,B.left) && isSub(A.right,B.right);
}
}