题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
关键词:二叉树
思路
子树
一棵大树 A,一棵小树 B,若 B 是 A 的子树,则:
- B 和 A 的结点值完全相同,它们俩的左子树、右子树所有结点的值也完全相同
- 或者 B 的左孩子和 A 的结点值完全相同,它们俩的左子树、右子树所有结点的值也完全相同
- 或者 B 的右孩子和 A 的结点值完全相同,它们俩的左子树、右子树所有结点的值也完全相同
子结构
子结构就是不用那么严格,图中的小框就是整棵树的子结构,图中的黄色大框也是整棵树的子结构,所以只要找到符合树的一部分树结点即可
所以可以编写函数,判断root2是否为root1的子树/子结构,区别在于子树是结点的直接分支结构,根结点必须是母对象的左结点或者右结点。而子结构可以是间接分支结构
代码实现
子树
链接:https://www.nowcoder.com/questionTerminal/6e196c44c7004d15b1610b9afca8bd88?answerType=1&f=discussion
来源:牛客网
public class IsSubTree {
public boolean HasSubtree(TreeNode root1, TreeNode root2) {
if (root1 == null || root2 == null) {
return false;
}
return judgeSubTree(root1, root2) ||
judgeSubTree(root1.left, root2) ||
judgeSubTree(root1.right, root2);
}
//(顺序)函数用来实现判断两个root是否相等
private boolean judgeSubTree(TreeNode root1, TreeNode root2) {
//(选择)先排除root1为空,root2为空的特殊情况,两个结点的值不等的情况
if (root2 == null) {
return true;
}
if (root1 == null) {
return false;
}
if (root1.val != root2.val) {
return false;
}
//(顺序)排除特殊情况后,当root1=root2时,继续比较其左右结点
return judgeSubTree(root1.left, root2.left) &&
judgeSubTree(root1.right, root2.right);
}
}
子结构
只要找到符合树的一部分树结点即可
链接:https://www.nowcoder.com/questionTerminal/6e196c44c7004d15b1610b9afca8bd88?answerType=1&f=discussion
来源:牛客网
public class Solution {
public boolean HasSubtree(TreeNode root1, TreeNode root2) {
if (root1 == null || root2 == null) {
return false;
}
return judgeSubTree(root1, root2) ||
judgeSubTree(root1.left, root2) ||
judgeSubTree(root1.right, root2);
}
private boolean judgeSubTree(TreeNode root1, TreeNode root2) {
if (root2 == null) {
return true;
}
if (root1 == null) {
return false;
}
//(选择)排除特殊情况时,子结点相对宽泛,当根结点不相同时,继续比较左结点与根结点
if (root1.val != root2.val) {
return judgeSubTree(root1.left, root2) ||
judgeSubTree(root1.right, root2);
}
return judgeSubTree(root1.left, root2.left) &&
judgeSubTree(root1.right, root2.right);
}
}
可以发现,子结构相对于子树,只是再judgeSubTree函数中,
子树,当根结点不同时,直接返回false,即不是子树
if (root1.val != root2.val) {
return false;
}
子结构,当根结点比较不相同时,给了继续比较的可能
if (root1.val != root2.val) {
return judgeSubTree(root1.left, root2) ||
judgeSubTree(root1.right, root2);
}