牛客网-剑指Offer-树的子结构

题目链接：https://www.nowcoder.com/practice/6e196c44c7004d15b1610b9afca8bd88?tpId=13&tqId=11170&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

 

先特判空树情况，然后使用递归的方法。

一、如果A和B的当前节点值相等，可以有三种选择：

　　1. A、B树均继续遍历左右节点，且两树左节点和右节点均必须相等。

　　2. B树回到根节点，继续遍历A的左子树。

　　3. B树回到根节点，继续遍历B的右子树。

 

二、如果当前节点值不相等，有两种选择，即一中的2、3

　　1. B树回到根节点，继续遍历A的左子树。
　　2. B树回到根节点，继续遍历B的右子树。

 

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {
public:
    bool dfs(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot2 == NULL)
            return true;
        
        if(pRoot1 == NULL && pRoot2 != NULL)
            return false;
        
        if(pRoot1->val == pRoot2->val)
            return dfs(pRoot1->left, pRoot2->left) && dfs(pRoot1->right, pRoot2->right)
            || dfs(pRoot1->left, pRoot2) || dfs(pRoot1->right, pRoot2);
        
        return dfs(pRoot1->left, pRoot2) || dfs(pRoot1->right, pRoot2);
    }
    
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(pRoot1 == NULL || pRoot2 == NULL)
            return false;
        return dfs(pRoot1, pRoot2);
    }
};