牛客网 - 剑指offer - “树的子结构”

题目描述

输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {
    }
};*/

class Solution {
public:
    bool isSubTree(TreeNode* p1, TreeNode* p2)
    {
        if(!p2) return true;    // p2已经遍历到底了还是对的，这部分就是对的
        if(p1 && p2 && p1->val == p2->val)
            return isSubTree(p1->left, p2->left) && isSubTree(p1->right, p2->right);
        else
            return false;
    }
    
    bool HasSubtree(TreeNode* p1, TreeNode* p2)
    {
        if(!p1 || !p2) return false;
        return isSubTree(p1, p2) || HasSubtree(p1->left, p2) || HasSubtree(p1->right, p2);
    }
};