问题描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
思路:
- 首先实现一个函数, 判断两棵树的结构是否相同
- 从数A的根节点开始, 递归左子树, 递归右子树, 逐个与树B进行比较, 当两棵树结构相同时, 说明树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 JudgeSameTree(TreeNode* pRoot1, TreeNode* pRoot2)
{
//当 pRoot2为NULL时, 说明之前的节点全部满足(即pRoot1->val == pRoot2->val)
if (pRoot2 == nullptr)
return true;
if (pRoot1 == nullptr || pRoot2 == nullptr)
return false;
//节点值不同时, 直接返回false
if (pRoot1->val != pRoot2->val)
return false;
//相同则继续递归下去(用逻辑 && 表示, 必须同时满足左子树和右子树均结构相同)
return isSameTree(pRoot1->left, pRoot2->left) && isSameTree(pRoot1->right, pRoot2->right);
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
//任何树都不是空树的子树
if (pRoot1 == nullptr)
return false;
//空树不是任意树的子树
if (pRoot1 != nullptr && pRoot2 == nullptr)
return false;
//判断两棵树结构是否相同
if (isSameTree(pRoot1, pRoot2))
return true;
//不相同则递归左子树, 递归右子树(用逻辑 || 表示 只要左子树或右子树满足即返回真)
return HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2);
}
};