且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

剑指offer系列之十六:树的子结构

更新时间:2022-06-20 16:36:29

题目描述

输入两颗二叉树A,B,判断B是不是A的子结构。

这实际上二叉树遍历算法的一种应用,要在原二叉树中查找是否具有某课子树,只需要判断每个节点是否都在二叉树中是否出现即可。所以需要先判断头结点,只有头结点符合要求才继续比较其子树是否符合,一样依次从头结点开始比较直到其左右子树进行比较,如果都符合则说明B是A的子结构。下面是基于这种思路实现的代码(已被牛客AC):

package com.rhwayfun.offer;

public class SubTree {

    public class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;

        }

    }

    public boolean HasSubtree(TreeNode root1, TreeNode root2) {
        boolean hasSubTree = false;
        if(root1 != null && root2 != null){
            if(root1.val == root2.val){
                hasSubTree = nodesValEqual(root1,root2);
            }
            if(!hasSubTree){
                hasSubTree = HasSubtree(root1.left, root2);
            }
            if(!hasSubTree){
                hasSubTree = HasSubtree(root1.right, root2);
            }
        }

        return hasSubTree;
    }

    private boolean nodesValEqual(TreeNode root1, TreeNode root2) {
        if(root2 == null) return true;
        if(root1 == null) return false;
        if(root1.val != root2.val) return false;
        return nodesValEqual(root1.left, root2.left) && nodesValEqual(root1.right, root2.right);
    }
}