输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解题思路:第一步在树1中查找与根节点的值一样的节点,这实际上就是树的遍历。对于二叉树这种数据结构熟悉的人来说自然知道可以用递归的方法去遍历,也可以用循环的方法去遍历。由于递归的代码实现比较简洁,面试时如果没有特别要求,那么我们通常会采用递归的方式。
如果发现某一结点的值和树B的头节点的值相同,则调用递归进行第二部判断。
第二步是判断树1中的根节点的子树是不是和树2具有相同的结构,同样,我们也可以用递归的思路来考虑:如果节点R的值和树B的根节点不相同,则以根节点的子树和树2肯定不具有相同的节点,如果它们的值相同,则递归地判断它们各自的左右节点的值是不是相同。递归的终止条件是我们到达了树1或者树2的叶节点。
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
boolean result = false;
if(root1!=null&&root2!=null){
if(root1.val==root2.val){
result = HasSubtreeCore(root1, root2);
}
if(!result){
result = HasSubtree(root1.left, root2);
}
if (!result) {
result = HasSubtree(root1.right, root2);
}
}
return result;
}
private boolean HasSubtreeCore(TreeNode root1, TreeNode root2) {
// TODO Auto-generated method stub
if(root2==null){
return true;
}
if(root1==null){
return false;
}
if(root1.val!=root2.val){
return false;
}
return HasSubtreeCore(root1.left, root2.left)&&HasSubtreeCore(root1.right, root2.right);
}
}