输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
测试用例:
功能测试:
树1和树2都是普通的二叉树,树2是或者不是树1的子结构
特殊输入测试:
两颗二叉树的一个或者两个根节点为null、
二叉树的所有节点都没有左子树或者右子树
方法一:
思路:参考剑指offer
1、首先设置标志位result = false,因为一旦匹配成功result就设为true, 剩下的代码不会执行,如果匹配不成功,默认返回false
2、递归思想,如果根节点相同则递归调用DoesTree1HaveTree2(), 如果根节点不相同,则判断tree1的左子树和tree2是否相同, 再判断右子树和tree2是否相同
3、注意null的条件,HasSubTree中,如果两棵树都不为空才进行判断,DoesTree1HasTree2中,如果Tree2为空,则说明第二棵树遍历完了,即匹配成功,tree1为空有两种情况
(1)如果tree1为空&&tree2不为空说明不匹配,
(2)如果tree1为空,tree2为空,说明匹配。
package co.com.jianzhioffer;
public class Solution18 {
public class TreeNode {
int val;
TreeNode left = null;
TreeNode right = null;
TreeNode(int val) {
this.val = val;
}
}
/*
* 递归实现,方法一
* 思路:参考剑指offer 1、首先设置标志位result = false,因为一旦匹配成功result就设为true,
* 剩下的代码不会执行,如果匹配不成功,默认返回false 2、递归思想,如果根节点相同则递归调用DoesTree1HaveTree2(),
* 如果根节点不相同,则判断tree1的左子树和tree2是否相同, 再判断右子树和tree2是否相同
* 3、注意null的条件,HasSubTree中,如果两棵树都不为空才进行判断,
* DoesTree1HasTree2中,如果Tree2为空,则说明第二棵树遍历完了,即匹配成功,
* tree1为空有两种情况(1)如果tree1为空&&tree2不为空说明不匹配, (2)如果tree1为空,tree2为空,说明匹配。
*/
public boolean HasSubtree(TreeNode root1, TreeNode root2) {
boolean result = false;
if (root1 != null && root2 != 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 boolean DoesTree1HaveTree2(TreeNode root1, TreeNode root2) {
if (root2 == null)
return true;
if (root1 == null)
return false;
if (root1.val != root2.val)
return false;
return DoesTree1HaveTree2(root1.left, root2.left)
&& DoesTree1HaveTree2(root1.right, root2.right);
}
}
方法二:上面方法的简洁版,利用好短路特性,不需要那么多flag
public boolean HasSubtree(TreeNode root1, TreeNode root2) {
if(root1 ==null || root2 == null) return false;
return isSubtree(root1, root2) || HasSubtree(root1.left, root2) || HasSubtree(root1.right, root2);
}
public boolean isSubtree(TreeNode root1, TreeNode root2){
if(root2 == null) return true;
if(root1 == null) return false;
if(root1.val == root2.val)
return isSubtree(root1.left, root2.left) && isSubtree(root1.right, root2.right);
else return false;
}