剑指Offer对答如流系列 - 树的子结构

Posted 2021-03-13 jefferychenxiao

面试题26：树的子结构

问题描述

输入两棵二叉树A和B，判断B是不是A的子结构。二叉树的定义如下：

    public class TreeNode{
        double val;
        TreeNode left = null;
        TreeNode right =null;
        public TreeNode(int val) {
            this.val=val;
        }
    }

比如下面的 B是A的子结构

问题分析

看了看《剑指Offer》高质量代码章节的面试题，发现难度都不高，但是没有分析好边界条件亦或是想当然就是容易出错，细心从来不是说说而已。请重视自己代码的规范性、完整性和鲁棒性。

这道题的思路可以概况如下：

1. 先对A树进行遍历，找到与B树的根结点值相同的结点R；

2. 判断A树中以R为根结点的子树是否包含B树一样的结构。

问题解答

    // 方法入口，对每个结点遍历判断
    public boolean hasSubtree(TreeNode root1,TreeNode root2) {
        if(root1==null || root2==null) {
            return false;
        }
        return doesTree1HasTree(root1, root2)|| hasSubtree(root1.left, root2)
                ||hasSubtree(root1.right, root2);
    }

    // 判断root结点开始的子树中各个结点是否相同
    private boolean doesTree1HasTree(TreeNode root1,TreeNode root2) {
        if(root2==null) return true;
        if(root1==null) return false;
        return equal(root1.val, root2.val) && doesTree1HasTree(root1.left, root2.left)
                && doesTree1HasTree(root1.right, root2.right);
    }

    // 判断两个浮点数是否相等
    private boolean equal(double num1,double num2) {
        return num1 - num2 < 0.0000001 && num1 - num2 > -0.0000001;
    }