题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
树的结构:
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
解题思路:
1.对于两棵二叉树来说,要判断B是不是A的子结构,首先第一步在树A中查找与B根节点的值一样的节点。
2.通常对于查找树中某一个节点,我们都是采用递归的方法来遍历整棵树。
3.第二步就是判断树A中以R为根节点的子树是不是和树B具有相同的结构。
4.这里同样利用到了递归的方法,如果节点R的值和树的根节点不相同,则以R为根节点的子树和树B肯定不具有相同的节点;
5.如果它们值是相同的,则递归的判断各自的左右节点的值是不是相同。
6.递归的终止条件是我们达到了树A或者树B的叶节点。
解题代码:
1.利用递归遍历整棵树中的所有结点,分别拿这些结点与B子树根节点进行比较,如果相同的话再以A子树中的那个结点作为根节点,分别比较它与B子树的左右子树。
public static boolean HasSubtree(TreeNode root1, TreeNode root2) {
boolean result = false;
if (root2 != null && root1 != null) {
//如果找到了对应TreeB的根节点的点
if(root1.val == root2.val){
//以这个根节点为为起点判断是否包含TreeB
result = doesTree1HaveTree2(root1,root2);
}
//如果找不到,那么就再去root的左儿子当作起点,去判断时候包含Tree2
if (!result) {
result = HasSubtree(root1.left,root2);
}
//如果还找不到,那么就再去root的右儿子当作起点,去判断时候包含Tree2
if (!result) {
result = HasSubtree(root1.right,root2);
}
}
//返回结果
return result;
}
2.在根节点相等的情况下,分别比较他们的左右子树,如果有某个值不相等,那就返回false。如果在比较完所有的结点之后没有发现数值不相等时,那就返回true。
public static boolean doesTree1HaveTree2(TreeNode node1, TreeNode node2) {
//如果TreeB已经遍历完了都能对应的上,返回true
if (node2 == null) {
return true;
}
//如果TreeB还没有遍历完,TreeA却遍历完了。返回false
if (node1 == null) {
return false;
}
//如果其中有一个结点的值不相等,返回false
if (node1.val != node2.val) {
return false;
}
//如果根节点对应的上,那么就分别去子节点里面匹配
return doesTree1HaveTree2(node1.left,node2.left) && doesTree1HaveTree2(node1.right,node2.right);
}
完整代码:
public static boolean HasSubtree(TreeNode root1, TreeNode root2) {
boolean result = false;
if (root2 != null && root1 != null) {
if(root1.val == root2.val){
result = doesTree1HaveTree2(root1,root2);
}
if (!result) {
result = HasSubtree(root1.left,root2);
}
if (!result) {
result = HasSubtree(root1.right,root2);
}
}
return result;
}
public static boolean doesTree1HaveTree2(TreeNode node1, TreeNode node2) {
if (node2 == null) {
return true;
}
if (node1 == null) {
return false;
}
if (node1.val != node2.val) {
return false;
}
return doesTree1HaveTree2(node1.left,node2.left) && doesTree1HaveTree2(node1.right,node2.right);
}