对于两棵彼此独立的二叉树A和B,请编写一个高效算法,检查A中是否存在一棵子树与B树的拓扑结构完全相同。
给定两棵二叉树的头结点A和B,请返回一个bool值,代表A中是否存在一棵同构于B的子树。
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
class IdenticalTree {
public:
bool cmp_tree(TreeNode *a, TreeNode *b)
{
if(a != NULL && b == NULL)
return false;
if(a == NULL && b == NULL)
return true;
if(a == NULL && b != NULL)
return false;
if(a->val != b->val)
return false;
return cmp_tree(a->left, b->left) &&
cmp_tree(a->right, b->right);
}
bool chkIdentical(TreeNode* A, TreeNode* B) {
// write code here
if(A==NULL && B==NULL)
return true;
if(A==NULL && B!=NULL)
return false;
return A != NULL &&(cmp_tree(A, B) || chkIdentical(A->left, B) ||
chkIdentical(A->right, B));
}
};
两个的区别在于第一个要求结构相同,第二个只需要包含即可。
输入两棵二叉树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 HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool result = false;
if(pRoot1 && pRoot2)
{
if(pRoot1->val == pRoot2->val)
result = DoesHsaTree(pRoot1,pRoot2)||HasSubtree(pRoot1->left,pRoot2)||HasSubtree(pRoot1->right,pRoot2);
}
return result;
}
bool DoesHsaTree(TreeNode* pRoot1,TreeNode* pRoot2)
{
if(!pRoot2)
return true;
if(!pRoot1)
return false;
if(pRoot1->val !=pRoot2->val)
return false;
return DoesHsaTree(pRoot1->left,pRoot2->left) && DoesHsaTree(pRoot1->right,pRoot2->right);
}
};