leetcode & 剑指 Offer 26. 树的子结构-优快云博客

本文链接：https://blog.youkuaiyun.com/u011026968/article/details/128753179

文章介绍了如何在C++中利用深度优先搜索(DFS)解决力扣上的一个问题，即判断一个树是否包含另一个树的子结构。关键在于检查节点值相等时，是否需要连续（由check参数控制）。代码展示了两种不同的DFS实现方式。

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

两种写法

1、一个Dfs

check的作用是：比如B节点是1 2 不用check的话可能只要A包含了1和2节点，哪怕不是连着的，也会返回true

class Solution {
public:
    bool isSubStructure(TreeNode* A, TreeNode* B) {
        if( A && !B) return false;
        
        return sub(A, B, 0);
    }

    bool sub(TreeNode* A, TreeNode* B, int check) {
        if(!B) return true;
        if(!A) return false;
        
        if (A->val == B->val) {
            if(check) return (sub(A->left, B->left, 1) && sub(A->right, B->right, 1));
            else
                return (sub(A->left, B->left,1) && sub(A->right, B->right,1)) || (
                sub(A->left, B,0) || sub(A->right, B,0)
            ) ;
        } else {
            if(check) return false;
            return sub(A->left, B, 0) || sub(A->right, B, 0);
        }
    }
};

2、两个dfs

class Solution {
public:
    bool isSubStructure(TreeNode* A, TreeNode* B) {
        if( A && !B) return false;
        
        return sub(A, B, 0);
    }

    bool sub(TreeNode* A, TreeNode* B, int check) {
        if(!B) return true;
        if(!A) return false;
        
        if (A->val == B->val) {
            if(check) return (sub(A->left, B->left, 1) && sub(A->right, B->right, 1));
            else
                return (sub(A->left, B->left,1) && sub(A->right, B->right,1)) || (
                sub(A->left, B,0) || sub(A->right, B,0)
            ) ;
        } else {
            if(check) return false;
            return sub(A->left, B, 0) || sub(A->right, B, 0);
        }