题目描述:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解:
可以想到遍历每一个B中的元素是否能够在A中找到,如果每一个元素都能在A中一一对应找到那B就是A的子树。因此可以采用递归的方法就行一一匹配。代码如下:
第一步:只有在左右树的根节点不为空的情况下才会进行比对。临时变量result作为返回值。
如果左右根节点都不为空,就进行比对,如果左右根节点的值相同,那就可以递归的进行左右子树的比对了。
如果左右根节点的值不同,那就直接返回false。
如果左右根节点都为空,直接返回false。
第二步:递归函数里面,递归终止条件有三个:
如果B树元素全部遍历完成,则说明B树是A的子树,直接返回true。
如果B树不为空,如果A树此时已经为空,说明B树不是A的子树,返回false。
如果上述两个条件都不满足,而且此时左右节点的值不相同,说明B树也不是A的子树,返回false。
第三步:如果第二步中的三个条件都不满足,说明已经遍历过的元素中B树是A的子树,可以继续递归遍历A的左右子树和B的左右子树进行匹配。
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
boolean result = false;
if(root1 != null && root2 != null){
if(root1.val == root2.val)
result = Contain(root1,root2);
if(result == false)
result = HasSubtree(root1.left,root2);
if(result == false)
result = HasSubtree(root1.right,root2);
}
return result;
}
public boolean Contain(TreeNode root1,TreeNode root2){
if(root2 == null)
return true;
if(root1 == null)
return false;
if(root1.val != root2.val)
return false;
return Contain(root1.left,root2.left)&&Contain(root1.right,root2.right);
}