题目
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解题思路
以下是使用的递归思路:
第一步:如果root2为空,则说明root2是root1的子结构。(这需要一个辅助条件,就是这题目上要求,root2为空即返回false需要排除掉) 如果root1为空,则说明不是root1的子结构。
第二步:如果root1和root2不相同,则返回root1的子节点和root2的比较。
第三步:如果root1和root2相同,则比较他们的子节点是否相同,如果相同返回true,如果不相同,则返回root1的子节点和root2的比较。
代码如下
//树的子结构
public boolean HasSubtree(TreeNode root1, TreeNode root2) {
if (root2 == null) return false;
return helper(root1, root2);
}
public boolean helper(TreeNode root1, TreeNode root2) {
if (root2 == null) return true;
if (root1 == null) return false;
if (root1.val == root2.val) {
if(helper(root1.left, root2.left)
&& helper(root1.right, root2.right)){
return true;
}
}
return helper(root1.left, root2) || helper(root1.right, root2);
}