题目:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
思路:
刚开始考虑将两个树的值分别输出到ArrayList中,然后看list1是否包含list2.
这种解法虽然可以通过牛客网的测试,但会出现偶然性(包含元素但不是在一个子树或顺序不一致)
利用递归的思路进行解题,分别比较二者的值,左子树和右子树.如果当两者出现不一致或者为null的时候返回false
代码:
/**
* 思路:
* 1.如果节点其中一个为null,直接返回false
* 2.先判断r1,r2的根节点,如果不满足在依次判断r1.left和r1.right
*/
public class HasSubTree {
public boolean HasSubtree(TreeNode root1, TreeNode root2) {
if(root1 == null || root2 == null)
return false;
return isSubTree(root1,root2)||
HasSubtree(root1.left,root2)||
HasSubtree(root1.right,root2);
}
public boolean isSubTree(TreeNode root1 , TreeNode root2){
if(root2 == null)
return true;
if(root1 == null)
return false;
if(root1.val == root2.val){
return isSubTree(root1.left,root2.left)&&
isSubTree(root1.right,root2.right);
}
return false;
}
}
网友评论