题目链接:https://www.nowcoder.com/questionTerminal/6e196c44c7004d15b1610b9afca8bd88
题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
第一次编辑代码:
/**
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) {
if(root2 == null){
return false;
}
if(root1 == null){
return false;
}
if(compare(root1,root2) == true){
return true;
}else{
return HasSubtree(root1.left,root2)||HasSubtree(root1.right,root2);
}
}
public boolean compare(TreeNode root1,TreeNode root2) {
if(root2 == null){
return true;
}
if(root1.val != root2.val){
return false;
}else{
return compare(root1.left,root2.left)&&compare(root1.right,root2.right);
}
}
}
提交结果
答案错误:您提交的程序没有通过所有的测试用例
用例:
{8,#,8,#,9,#,2,#,5},{8,#,9,3,2}
对应输出应该为:
false
你的输出为:
java.lang.NullPointerException
反思
用例的表示法和我想的有些出入。
知道了,是a的子树为空时忘了做判断导致的异常。
第二次编辑代码:
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
if(root2 == null || root1 == null){
return false;
}
return compare(root1,root2)||HasSubtree(root1.left,root2)||HasSubtree(root1.right,root2);
}
public boolean compare(TreeNode root1,TreeNode root2) {
if(root2 == null){
return true;
}
if(root1 == null){
return false;
}
if(root1.val != root2.val){
return false;
}else{
return compare(root1.left,root2.left)&&compare(root1.right,root2.right);
}
}
}