牛客OJ：判断子结构和子树

最新推荐文章于 2022-11-21 12:53:55 发布

原创最新推荐文章于 2022-11-21 12:53:55 发布 · 209 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#牛客OJ

NewCoder OJ 专栏收录该内容

37 篇文章

订阅专栏

本文介绍了一种用于判断一棵树是否为另一棵树子树的算法。该算法通过递归比较两棵树的节点值来实现，首先检查当前节点是否相同，再分别检查左右子节点。若找到匹配根节点，则深入验证子树；若未找到则继续遍历原树。

判断子结构：

class Solution {
public:
    bool solve(TreeNode* a, TreeNode* b){
        if(b == NULL) return true;
        if(a == NULL) return false;
        if(a->val == b->val){
            return 
                solve(a->left,b->left)
                && 
                solve(a->right,b->right);
        }
        return false;
    }
    bool judge(TreeNode* a,TreeNode* b){
        if(a == NULL || b == NULL) return false;
        bool ans = false;
        if(a->val == b->val) ans = (ans || solve(a,b));
        if(ans) return true;
        ans = (ans || judge(a->left,b));
        ans = (ans || judge(a->right,b));
        return ans;
    }
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        return judge(pRoot1,pRoot2);
    }
};

判断子树：

bool solve(TreeNode* a, TreeNode* b){
    if(a == NULL && b == NULL) return true;
    if(a == NULL || b == NULL) return false;
    if(a->val == b->val){
        return 
            solve(a->left,b->left)
            && 
            solve(a->right,b->right);
    }
    return false;
}
bool judge(TreeNode* a,TreeNode* b){
    if(a == NULL || b == NULL) return false;
    bool ans = false;
    if(a->val == b->val) ans = (ans || solve(a,b));
    if(ans) return true;
    ans = (ans || judge(a->left,b));
    ans = (ans || judge(a->right,b));
    return ans;
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
    return judge(pRoot1,pRoot2);
}