检查子树
描述
有两棵非常大的二叉树:T1,有几万个节点;T2,有几万个节点。
设计一个算法,判断 T2 是否为 T1 的子树。
如果 T1 有一个节点 n,其子树与 T2 一模一样,则 T2 为 T1 的子树
也就是说,从节点 n 处把树砍断,得到的树与 T2 完全相同。
样例
输入:t1 = [1, 2, 3], t2 = [2]
输出:true
思路
递归:从t1的每个节点开始对比,判断是否与t2树相同。
设计一个dfs递归函数,用以判断t1、t2的子树的结构是否相同。
当前两个节点不相同,则向t1的左孩子和右孩子继续判断。
当前两节点的值相等,才能继续向下递归判别
代码
public class Main {
public static boolean checkSubTree(TreeNode t1, TreeNode t2) {
if (t1==null){
return false;
}
if (t2==null){
return true;
}
//因为不一定是从根节点开始,两棵树一模一样
//也可能是从t1的某个节点开始,两棵树完全相同了
return dfs(t1,t2)
|| checkSubTree(t1.left,t2)
|| checkSubTree(t1.right,t2);
}
private static boolean dfs(TreeNode t1, TreeNode t2) {
if (t2==null){
return true;
}
if (t1==null||t1.val!=t2.val){
return false;
}
//此时t1节点和t2节点值相同,再继续判断二者的子树是否相同
return dfs(t1.left,t2.left) || dfs(t1.right,t2.right);
}
}