输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
首先判断两棵树的根节点是否相同,如果相同就递归判断左右节点,如果根节点不相同,则判断左右节点和根节点进行判断,然后递归即可。
我们用两个例子来进行解释说明,首先判断树A和树B的根节点是否相同,明显不相同,则判断树A根节点的左右节点和树B的根节点进行比较,发现也不符合,则依次递归下去,注意判空即可。完整代码如下(很遗憾这次就只用了一种解决方法,后面有时间,研究出了新的算法,会分享出来的。)
public class MainHasSubtree {
public static void main(String[] args) {
TreeNode a1 = new TreeNode(1);
TreeNode a2 = new TreeNode(7);
TreeNode a3 = new TreeNode(3);
TreeNode a4 = new TreeNode(2);
TreeNode a5 = new TreeNode(6);
TreeNode a6 = new TreeNode(4);
TreeNode a7 = new TreeNode(5);
a1.left = a2;
a1.right = a3;
a2.left = a4;
a2.right = a5;
a4.left = a6;
a4.right = a7;
TreeNode b1 = new TreeNode(2);
TreeNode b2 = new TreeNode(4);
TreeNode b3 = new TreeNode(5);
b1.left = b2;
b1.right = b3;
boolean b = HasSubtree(a1, b1);
System.out.println(b);
}
public static boolean HasSubtree(TreeNode root1,TreeNode root2) {
if(null == root1 || null == root2){
return false;
}
//首先判断两棵树的根节点,如果递归接口为true
if(subTree(root1,root2)){
return true;
}
//否则依次递归即可
return HasSubtree(root1.left,root2) || HasSubtree(root1.right,root2);
}
public static boolean subTree(TreeNode root1,TreeNode root2){
//第二颗树为空,则说明是子树,则返回true
if(null == root2){
return true;
}
//如果第一颗树为空,第二颗树不为空,则不是子树
if(null == root1){
return false;
}
return (root1.val == root2.val) && subTree(root1.left,root2.left) && subTree(root1.right, root2.right);
}
}