题目:输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
先说说我的思路:
我总想着通过添加静态的int类型的flag作为辅助来判断应该是哪种情况,然后再根据情况具体分析,结果越弄越乱了。
最后我查看牛客网上的讨论,看到 fuyufjh 的代码,豁然开朗,我才发现,其实很多情况,明显可以简单粗暴的解决,干嘛要想那么多啊。
转化为Java代码是:
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
if(root1==null||root2==null){
return false;
}
// return isSubtree(root1,root2)||isSubtree(root1.left,root2)||isSubtree(root1.right,root2); 有bug,应该是下面的
//PS:感谢 索氧 和 莫比乌斯灯泡 两个同学指出问题。
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);
}else{
return false;
}
}
给大家的建议:
我希望大家以后遇到问题,不要想得太多,抓住本质,从本质上出发。当然我相信我自己的那种想法也可以解决,但是嘛好头疼啊!