树的子结构

输入两棵二叉树 A,B,判断 B 是不是 A 的子结构。我们约定空树不是任意一个树的子结构

基本思路:

  • 首先在 A 中找到与 B 的根结点的值一样的结点,可以使用递归遍历查找

  • 找到的话,判断 A 中该结点对应的子树是否与 B 相等,相等返回 true,方法结束,否则返回 false,方法继续执行

  • 继续在 A 中遍历查找与 B 的根结点的值一致的结点,找到的话重复第二步,如果不存在这样的结点,直接返回 false

public class Solution {

    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        boolean result = false;
        // 当 Tree1 和 Tree2 都不为 null 时,才进行比较。否则直接返回 false
        if(root2 != null && root1 != null) {
            // 如果找到了对应 Tree2 的根节点的点
            if(root1.val == root2.val) {
                // 以这个根节点为起点判断是否包含 Tree2
                result = doesTree1HaveTree2(root1, root2);
            }
            // 如果找不到,那么就再去找 root 的左儿子当起点,判断是否包含 Tree2
            if(!result) {
                result = HasSubtree(root1.left, root2);
            }
            // 如果还找不到,那么就再去找 root 的右儿子当起点,去判断是否包含 Tree2
            if(!result) {
                result = HasSubtree(root1.right, root2);
            }
        }
        // 返回结果
        return result;
    }

    public boolean doesTree1HaveTree2(TreeNode node1,TreeNode node2) {
        // 如果 Tree2 已经遍历完了都能对应的上,返回 true
        // 注意这里必须先判断 node2,因为 Tree1 和 Tree2 是同步比较的,此时 node1 可能为 null,也可能不为 null
        if(node2 == null) {
            return true;
        }
        //如果 Tree2 还没有遍历完,Tree1 却遍历完了,返回false
        if(node1 == null) {
            return false;
        }
        //如果其中有一个点没有对应上,返回 false
        if (node1.val != node2.val) {
            return false;
        }
        // 如果根节点对应的上,那么就分别去子节点里面匹配
        return doesTree1HaveTree2(node1.left,node2.left) && doesTree1HaveTree2(node1.right,node2.right);
    }
}

版权声明:程序员胖胖胖虎阿 发表于 2023年9月1日 上午1:16。
转载请注明:树的子结构 | 胖虎的工具箱-编程导航

相关文章

暂无评论

暂无评论...