树的子结构

题目描述

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

 

注意：不一定子结构B在A中一定是包含叶子节点的，有可能也是中间那部分，所以刚开始的判断条件出错。

 

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

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

    bool Check(TreeNode* t1, TreeNode* t2)
    {
        /*if(t1 == NULL && t2 == NULL)
            return true;
        else if((t1 == NULL && t2 != NULL) || (t1 != NULL && t2 == NULL))
            return false;*/
        if(t2 == NULL)
            return true;
        if(t1 == NULL)
            return false;
        if(t1->val != t2 ->val)
            return false;
        if(Check(t1->left, t2->left) && Check(t1->right,t2->right))
            return true;
        return false;
    }
};