题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
注:一碰到递归的题就有点蒙圈了,虽然大概知道思路是怎么样的,就是写不出代码。还是练得太少了,言归正传。
思路:
遍历二叉树最好的方法是递归;
isSubtree(TreeNode root1,TreeNode root2)的作用:
判断两棵树的当前结点是否相等,若相等则继续在该方法中进一步判断,若不相等则返回false。当两树的关系确实是子结构时则返回true。
主方法HasSubtree(TreeNode root1,TreeNode root2)中利用短路原理去返回结果。
当isSubtree返回true,则直接返回true,否则继续遍历root1并判断与root2是否是子结构关系
我简单说一下短路原理是什么:
Java中有比较运算符&&和||,这两个运算符的作用大家不知道的可以去查一下。
例如 A && B
Java中当程序运行到该语句时,先判断A是否正确,若A正确,则再判断B。否则无论B是否正确,都不会再去判断B的正确性。
再说说A || B
先判断A是否正确,若A正确,则直接判断该语句是正确的,而不用再判断B的正确性。当A不正确的时候才会继续判断B的正确性。
下面贴代码:
/**
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 || root1 == null){
return false;
}
return (isSubtree(root1,root2) ||
HasSubtree(root1.left,root2) ||
HasSubtree(root1.right,root2));
}
/*
判断两棵树的结点是否相等,若相等则进一步判断,不相等则返回false
*/
public boolean isSubtree(TreeNode root1,TreeNode root2){
if(root2 == null){
return true;
}
if(root1 == null){
return false;
}
if(root1.val == root2.val){
//如果前一个条件是正确的,则再判断后一个条件,
//若前一个条件是false,那么语句直接返回false
return (isSubtree(root1.left,root2.left) &&
isSubtree(root1.right,root2.right));
}
else{
return false;
}
}
}