[剑指 Offer]26 树的子结构 Java实现_剑指offer26树的子结构 java-优快云博客

本文链接：https://blog.youkuaiyun.com/BedivereAlter/article/details/111492018

本文介绍了一种算法，用于判断一棵二叉树是否为另一棵二叉树的子结构。通过递归方法实现，包括搜索起始节点及匹配子树结构的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

题目

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构，即 A中有出现和B相同的结构和节点值。

例如:
给定的树 A:

 3
/ \

4 5
/
1 2
给定的树 B：

4
/
1
返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

示例 1：

输入：A = [1,2,3], B = [3,1]
输出：false
示例 2：

输入：A = [3,4,5,1,2], B = [4,1]
输出：true

代码

OMG这道题写了好久…我先贴一下我原来写的代码
在写startMatch方法的时候…我想不到一个很好的方法把这三个结果给&&起来, search的话因为是||所以可以先立一个flag = false就好…但是&&的话就不能这么做, 就只好把这三个放在一个return里…

import sun.reflect.generics.tree.Tree;

public class _26 {
    public static void main(String[] args) {

    }

    public static boolean isSubStructure(TreeNode A, TreeNode B) {
        if (B == null){
            return false;
        }
        return startSearch(A, B);
    }

    public static boolean startSearch(TreeNode A, TreeNode B) {
        boolean flag = false;
        System.out.println(A.val);
        if (A.val == B.val) {
            if (startMatch(A, B)) {
                return true;
            }
        }
        if (A.left != null) {
            flag  = false || startSearch(A.left, B);
        }
        if (A.right != null) {
            flag  = false || startSearch(A.right, B);
        }
        return flag;
    }

    public static boolean startMatch(TreeNode A, TreeNode B) {
        if (A.val == B.val) {
            if ((A.left == null && B.left == null) && (A.right == null && B.right == null)) {
                return true;
            }
            else if (A.left != null && B.left != null){
                startMatch(A.left, B.left);
            }
            else if (A.right != null && B.right != null){
                startMatch(A.right, B.right);
            }
        }
        return false;
    }

    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }
}

正确而精简的写法

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public static boolean isSubStructure(TreeNode A, TreeNode B) {
        return startSearch(A, B);
    }


    public static boolean startSearch(TreeNode A, TreeNode B) {
        if (A == null || B == null) {
            return false;
        }
        return startMatch(A, B) || startSearch(A.left, B) || startSearch(A.right, B);
    }
    public static boolean startMatch(TreeNode A, TreeNode B) {
        if(B == null){
            return true;
        }
        if(A == null){
            return false;
        }
        return A.val == B.val && startMatch(A.left, B.left) && startMatch(A.right, B.right);
    }
}