剑指Offer: 树的子结构 (java代码实现)

题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)

解题思路
遍历A二叉树的每一个结点 如果A二叉树中以这个结点形成的子树 有任意一个和B子树结构一样 那么B就是A的子结构 我们可以用递归来求解

下面是代码实现

public class Solution {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        if(root2 == null || root1 == null) {
            return false;
        }
        return  NodeHasSubtree(root1,root2)|| HasSubtree(root1.left, root2) || 
        HasSubtree(root1.right, root2);
        
    }
     public boolean NodeHasSubtree(TreeNode root1,TreeNode root2) {
         if(root2 == null) {
             return true;
         }
         if(root1 == null) {
             return false;
         }
         
         if(root1.val == root2.val) {
             return NodeHasSubtree(root1.left, root2.left) &&
             NodeHasSubtree(root1.right, root2.right);
         }       
         return false;
    }
}

我们可以看到HasSubtree函数 所做的是 前序遍历每个结点 首先判断当前结点所形成的子树是否和B树一样 然后递归判断 左数和右树是否满足 条件 只要这3个情况有一个满足就代表B是A的子树。

那如何判断A树中以该节点为头节点的树是否和 B树相等呢
我们用了NodeHasSubtree函数来求解

如果A树的当前节点为空 B子树当前结点为空 return true
B子树当前节点不为空 return false
如果A树当前节点不为空 B子树当前节点为空 return true
B子树当前节点不为空 判断值是否相等 继续递归判断左子树右子树;
那么可以简化为
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/wmh1152151276/article/details/87929992