《剑指offer》刷题笔记：17-树的子结构

最新推荐文章于 2022-09-11 22:59:03 发布

原创最新推荐文章于 2022-09-11 22:59:03 发布 · 168 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

22 篇文章

订阅专栏

本文介绍了一种算法，用于判断一棵二叉树B是否是另一棵二叉树A的子结构。通过递归方法，首先检查两树根节点是否相等，然后分别判断A的左右子树是否包含B。

题目描述

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

代码实现

/**
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) {
        // 设置标志位 result = false
        // 匹配成功 result 就设为 true，剩下的代码不会执行
        // 匹配不成功，默认返回 false
        boolean result = false;
        // 当 Tree1 和 Tree2 均不为空时，进行比较
        if(root2 != null && root1 != null){
            // 如果在 Tree1 找到了对应 Tree2 根节点的节点，以这个节点为起点判断是否包含 Tree2
            if(root1.val == root2.val){
                result = doseTree1HaveTree2(root1,root2);
            }
            // 如果找不到，那么就以 root1 的左儿子为起点判断
            if(!result){
                result = HasSubtree(root1.left,root2);
            }
            // 如果还找不到，那么就以 root1 的右儿子为起点判断
            if(!result){
                result = HasSubtree(root1.right,root2);
            }
        }
        // 返回结果
        return result;
    }
    public boolean doseTree1HaveTree2(TreeNode node1,TreeNode node2){
        // 前两个 if 判断语句不能颠倒顺序，否则可能会判断错误
        // 如果 Tree2 遍历结束，且节点均相同，返回 true
        if(node2 == null){
            return true;
        }
        // 如果 Tree2 还未遍历结束， Tree1 却遍历结束了，则返回 false
        if(node1 == null){
            return false;
        }
        // 如果有某个节点不相同，返回 false
        if(node1.val != node2.val){
            return false;
        }
        // 如果根节点相同，继续匹配子节点
        return doseTree1HaveTree2(node1.left,node2.left) && doseTree1HaveTree2(node1.right,node2.right);
    }
}