剑指offer 572. 另一棵树的子树

72 阅读1分钟

572. 另一棵树的子树 - 力扣(LeetCode)

先比较根节点

image.png 根节点不相等,再拿A树的左子树与B树根节点相比较。

如果相等。

那么再比较C树的左子树是否和B树的左子树相同,C树的右子树是否和B树的右子树相同

image.png

code

class Solution {
public:
    bool compare(TreeNode* root, TreeNode* subRoot) {
        //如果两个都为空
        if(root==NULL&&subRoot==NULL){
            return true;
        }
        //如果只有一个为空
        if(root==NULL||subRoot==NULL){
            return false;
        }
        //如果两个都不空,结点值也不同,那直接返回false
        if(root->val!=subRoot->val){
            return false;
        }
        //如果现在结点值和子树结点值相同,再分别检查两个的左右孩子
        return compare(root->left, subRoot->left) && compare(root->right, subRoot->right);

    }
    bool isSubtree(TreeNode* root, TreeNode* subRoot) {
        //如果要检查的子树为空,那么不用查了,肯定对的
        if(subRoot == NULL) return true;

        //如果要检查的子树不空,但root是空的,那也不用查了,错的。
        if(root == NULL) return false;

        //要么是它本身,要么是它左子树,要么是它的右子树
        return compare(root, subRoot) || isSubtree(root->left, subRoot) || isSubtree(root->right, subRoot);
    }
};