问题:输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
思路:1、判断B是不是空树,如果是空树,返回false;
2、判断A是不是空树,如果是空树,返回false;
3、如果找到了对应B的根节点的点,即在这时候AB相同了,以这个根节点为为起点判断是否包含B;
4、判断是否包含B?-->因为此时找到了AB相同的节点,所以要遍历他们的左右节点是否相同,结束条件是B为空,如果B为空,说明匹配完成了,B是子树,返回true;如果B还没结束,A先结束了,说明不匹配,返回false;如果匹配过程中value值不相等,返回false;
5、如果不包含,则用A根节点的左儿子判断是否包含B,还不包含,就用A根节点的右儿子判断是否包含B。
代码:
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
if(root2 == null){
return false;
}
if(root1 == null){
return false;
}
boolean result = false;
if(root1.val == root2.val){
result = ContainSubtree(root1,root2);
}
if(!result){
result = HasSubtree(root1.left,root2);
if(!result){
result = HasSubtree(root1.right,root2);
}
}
return result;
}
public boolean ContainSubtree(TreeNode root1,TreeNode root2){
if(root2 == null){
return true;
}
if(root1 == null && root2 != null){
return false;
}
if(root1.val != root2.val){
return false;
}
return (ContainSubtree(root1.left,root2.left) && ContainSubtree(root1.right,root2.right));
}
}