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

文章目录

面试题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;
    }