17.树的子结构
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
树A中是否存在和树B结构一样的子树,可以分成两步:
第一步在树A中找到和B的根节点的值一样的结点R;
第二步再判断树A中以R为根结点的子树是不是包含和树B一样的结构。
用HasSubtree来寻找节点R。
- 如果找到A和B根节点的值一样时,则需要比较他们的左右子树,只要有一个满足与被比较结点是有相同的子树,则返回true
- 如果值不相同直接返回false;
用isSubtree判断A和B是否是相同子树。
当B=null时说明能够遍历到叶子,则B是A子树。
当A=null&&B!=null则B还没遍历完,A就结束了,那么肯定不是子树。
public class Solution {
public boolean HasSubtree(TreeNode rootA,TreeNode rootB) {
if(rootB==null) return false;
if((rootA==null)&&(rootB!=null)) return false;
// 在两棵树中,只要两个结点有一个是空结点,不需要比较,直接返回false。
boolean flag=false;//首先设置标志位result = false,因为一旦匹配成功result就设为true,
//如果这两个结点A,B的值相同,则要比较他们的左右子树,只要有一个子树不同那么就没有父子关系。
if(rootA.val==rootB.val){以这个根节点为为起点判断是否包含树B
flag=isSubtree(rootA,rootB);
}
//如果找不到,那么就再去root的左儿子当作起点,去判断时候包含树B
if(!flag){
flag=HasSubtree(rootA.left,rootB);
}
//如果还找不到,那么就再去root的右儿子当作起点,去判断时候包含树B
if(!flag){
flag=HasSubtree(rootA.right,rootB);
}
return flag;
}
boolean isSubtree(TreeNode prootA,TreeNode prootB){
if(prootB == null) return true;
if(prootA==null) return false;
//如果其中有一个点没有对应上,返回false
if(prootA.val!=prootB.val) return false;
//如果根节点对应的上,那么就分别去子节点里面匹配
return isSubtree(prootA.left,prootB.left)&& isSubtree(prootA.right,prootB.right);
}
}