17. 树的子结构

/* 树的子结构*/
/*输入两棵二叉树A，B，判断B是不是A的子结构。PS：我们约定空树不是任何树的子结构*/
/*思路：首先考虑边界条件：如果树A为空或者树B为空，那么返回值都应为false；
然后，判断两棵树的根结点是否相等，如果相等的话，再判断树的左子树以及右子树是否分别相等；
如果一开始两棵树的根结点不相等，那么递归调用函数，判断树A中的左子树以及右子树中是否有跟
树B的根结点相等的结点。
*/

/*
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)
    {
        bool result = false;
        if(pRoot1 != NULL && pRoot2 != NULL){
            if(pRoot1->val == pRoot2->val)
                result = DoesRoot1HasRoot2(pRoot1, pRoot2);
            if(!result)
                result = HasSubtree(pRoot1->left, pRoot2);
            if(!result)
                result = HasSubtree(pRoot1->right, pRoot2);
        }
        return result;
    }
    bool DoesRoot1HasRoot2(TreeNode* root1, TreeNode* root2){
        if(root2 == NULL)
            return true;
        if(root1 == NULL)
            return false;
        if(root1->val != root2->val)
            return false;
        return DoesRoot1HasRoot2(root1->left, root2->left) && DoesRoot1HasRoot2(root1->right, root2->right); 
    }
};