输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
// 树节点
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
// 比较树
bool compareTree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool ret = false;
do
{
if (pRoot1 == NULL && pRoot2 == NULL) {
ret = true;
break;
}
if (pRoot2 == NULL) { // 子树遍历到最后
ret = true;
break;
}
if (pRoot1 != NULL && pRoot2 != NULL) {
if (pRoot1->val == pRoot2->val) {
if (!compareTree(pRoot1->left, pRoot2->left)) { // 左边不相等
break;
}
if (!compareTree(pRoot1->right, pRoot2->right)) { // 右边不相等
break;
}
// 全部相等
ret = true;
}
}
} while (0);
return ret;
}
// 是否是子树
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool ret = false;
do
{
if (NULL == pRoot1 || NULL == pRoot2) {
break;
}
if (pRoot1->val == pRoot2->val && compareTree(pRoot1, pRoot2)) { // 值相等 且全部比较相等
ret = true;
break;
}
if (HasSubtree(pRoot1->left, pRoot2)) { // 左边找到
ret = true;
break;
}
if (HasSubtree(pRoot1->right, pRoot2)) { // 右边找到
ret = true;
break;
}
} while (0);
return ret;
}
测试