树的子结构笔记

最新推荐文章于 2024-04-25 12:28:07 发布

原创最新推荐文章于 2024-04-25 12:28:07 发布 · 223 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#剑指offer

剑指offer 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种算法，用于判断一棵二叉树B是否为另一棵二叉树A的子结构。通过递归方法实现了核心函数doesTree1HaveTree2，该函数检查树B是否完全匹配树A中的部分结构。

描述

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

思路

数A任何一个节点都有可能是树B的头结点，因此需要遍历树A的所有节点。需要些一个函数来判断当前节点是否等于树B，
写一个布尔类型函数doesTree1HaveTree2，参数为树A头结点，和树B头结点。



public class Solution {
    public static boolean HasSubtree(TreeNode root1, TreeNode root2) {
        boolean result = false;
        if (root2 != null && root1 != null) {
            if(root1.val == root2.val){
                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 static boolean doesTree1HaveTree2(TreeNode node1, TreeNode node2) {
        if (node2 == null) {
            return true;
        }
        if (node1 == null) {
            return false;
        }
        //如果其中有一个点没有对应上，返回false
        if (node1.val != node2.val) {  
                return false;
        }

        //如果根节点对应的上，那么就分别去子节点里面匹配
        //这里也是一个递归的比较
        return doesTree1HaveTree2(node1.left,node2.left) && doesTree1HaveTree2(node1.right,node2.right);
    }