LeetCode 572. 另一棵树的子树

题目

题解

• 注意是所有后代节点不是包含关系

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    // bool issame(TreeNode* p, TreeNode* q){
    //     if(!p&&!q)
    //         return true;
    //     if(q&&q&&p->val==q->val)
    //         return issame(p->left,q->left)&&issame(p->right,q->right);
    //     return false;
    // }
    bool issame(TreeNode* p, TreeNode* q){
        if(!p&&!q)
            return true;
        if((p && !q) || (!p && q) || (p->val != q->val))
            return false;
        return issame(p->left,q->left)&&issame(p->right,q->right);
    }


    bool isSubtree(TreeNode* root, TreeNode* subRoot) {
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty()){
            auto node=que.front();
            que.pop();
            if(node->val==subRoot->val){
                if(issame(node,subRoot)) return true;
            }
            if(node->left) que.push(node->left);
            if(node->right) que.push(node->right);
        }
        return false;
    }
};