题目
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
思路
这题要借助递归的思想,边界一定要判断清楚,不然就很容易出错。
想法简单,先判断两颗树根节点是不是相同,如果相同,就判断树1的根节点开始的位置下是不是包含子树2。如果根节点不同,那从树1的左子树判断是不是包含子树2。如果还没找到,就从树1的右子树判断是不是包含子树2。
其中 从 树的某位置 判断 是否包含子树 这个过程要作为另一个递归函数。
判断子树的递归函数,要注意的地方很多:
要先判断子树2是否遍历完都找到了,则返回true
如果子树2还没完,树1完了,就要返回false
注意!!
这里一定要先判断子树2,然后判断树1.
如果先判断树1是否遍历完,树1遍历完,找到了树2为Ture没找到树2为FALSE,没法判断return的值。但树2一旦遍历完了,必定表示找到该子树,要return true,这点一定要考虑清楚,绝对不能调换位置。
/**
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) {
boolean result=false;
if(root1!=null&&root2!=null)
{
if(root1.val==root2.val)
result=doeshassubtree(root1,root2);
if(!result)
result=HasSubtree(root1.left,root2);
if(!result)
result=HasSubtree(root1.right,root2);
}
return result;
}
public boolean doeshassubtree(TreeNode node1,TreeNode node2) {
if(node2==null)
return true;
if(node1==null)
return false;
if(node1.val==node2.val)
return doeshassubtree(node1.left,node2.left)&&doeshassubtree(node1.right,node2.right);
else
return false;
}
}