版权声明:本文为博主原创文章,未经博主允许不得转载。
题目描述:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)。
解题思路:
首先看牛客网给出的测试用例:
一般对于树的操作不像链表一样,操作更复杂,如果使用循环遍历的话,对于非完全二叉树规律难寻,一般通用的方法就是使用递归求解,本题也不例外,同样使用递归求解,求解的大体思路是首先判断B的根节点和A的根节点是否相同(这里的相同是指节点的值相同并且左右子节点相同),如果相同比较他们的左右子节点,这一步骤是相同的,可以用递归完成,直到B遍历到每个尾节点,如果这一过程比较的所有节点是相同的,则证明B是A的子结构。如果B的根节点和A的根节点不同,则A向他的左右子节点滑动,然后继续跟B的子节点比较,步骤同上。
递归的使用可以用我总结的三步来完成。求解过程如下:
1. 递归截止条件。
递归的截止条件是为了递归能够避免无限循环下去,首先来分析什么情况下递归截止返回遍历结果,(1)根据题目要求,如果B数是个空树,递归截止。(2)如果被遍历的树A是空树,自然而然递归截止。(3)如果比较的是B的尾节点,无法进行下去,递归也会截止。(4)如果A树从头遍历到尾始终没有和B的节点相同的节点,递归截止。
2. 递归的前后衔接。
如果A的根节点值以及左右子节点情况和B的根节点完全相同,那么A和B都继续滑动到他们的左右子节点进行比较;如果A的根节点值和B的根节点值是相同的,但是左右子节点的情况是不相同的,那么只滑动A到他的左右子节点再与B比较。如果A的根节点的值和B的根节点的值不相同,那么A直接滑动到他的左右自己点再和B的根节点比较,直到遍历完成。
3. 递归节点数据的处理。
根据题目,本题目中用到的递归并没有数据处理,只是比较判断两个树节点是否相同。对于其他递归,可以具体情况具体对待。
源码:
/**
* 输入两棵二叉树A,B,判断B是不是A的子结构。
* @param root1 A树
* @param root2 B数
* @return
*/
public static boolean HasSubtree(TreeNode root1,TreeNode root2) { if (root2==null) { //空树不是任意一个树的子结构
return false;
}
if (root1==null) { //如果A为空,那肯定返回false
return false;
}
if(root2.val==root1.val){ //A和B比较的根节点的值相同 if (root2.left==null&&root2.right==null) { //比较的节点是B的尾节点,递归截止
return true;
}
//下面三种比较的是比较的节点完全相同的情况
if ((root2.left!=null&&root2.right!=null)&&(root1.left!=null&&root1.right!=null)&&root2.left.val==root1.left.val&&root2.right.val==root1.right.val) {
return HasSubtree(root1.left,root2.left)&& HasSubtree(root1.right,root2.right);
}else if ((root2.left!=null&&root2.right==null)&&(root1.left!=null&&root1.right==null)&&root2.left.val==root1.left.val) {
return HasSubtree(root1.left,root2.left);
}else if ((root2.left==null&&root2.right!=null)&&(root1.left==null&&root1.right!=null)&&root2.right.val==root1.right.val) {
return HasSubtree(root1.right,root2.right);
}else{ //比较的节点不同,A向左右子节点移动一个再比较
if (root1.left!=null&&root1.right!=null) {
return HasSubtree(root1.left,root2)|| HasSubtree(root1.right,root2);
}else if(root1.left==null&&root1.right!=null){
return HasSubtree(root1.right,root2);
}else if(root1.left!=null&&root1.right==null){
return HasSubtree(root1.left,root2);
}else {
return false;
}
}
//比较的根节点的值不相同,直接向左右子节点滑动
}else if(root1.left!=null&&root1.right==null){
return HasSubtree(root1.left,root2);
}else if(root1.left==null&&root1.right!=null){
return HasSubtree(root1.right,root2);
}else{
return false;
}
}