牛客链接
题目
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解题思路
什么是子结构?
在看牛客时概念有些混乱,子树和子结构是不同的。
子结构可以是原树的任意一个节点的结构相同即可(B树在A树种出现即可)。
什么是子树呢?
子树代表这棵二叉树必须是原树的本树或左子树或右子树(B树必须是A树的左子树,右子树,或者和A树相同)
思路
可以看到,本题是要求B树是A树的子结构而非子树,那么,只要B树存在于A中即可,需要遍历A全部节点,而不仅仅是根节点和根节点的左右孩子节点。
我们就需要遍历A树,在A树中找到节点值和B树的根节点值相同的节点,然后判断该节点的结构和B树的结构是否相同,如果相同,那么将结果改为true,如果不相同,就继续搜寻遍历A树,直到找到为止。
代码实现
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
if(root1 == null || root2 == null) return false;
boolean res = false;
if(root1.val == root2.val){
res = HasSubtreeRuput(root1,root2);
}
if(!res) {
res = HasSubtree(root1.left,root2);
}
if(!res) {
res = HasSubtree(root1.right,root2);
}
return res;
}
public boolean HasSubtreeRuput(TreeNode root1,TreeNode root2){
if(root2 == null) return true;
if(root1 == null) return false;
if(root1.val!=root2.val) return false;
return HasSubtreeRuput(root1.left,root2.left)&&HasSubtreeRuput(root1.right,root2.right);
}
}