核心考点:二叉树理解,二叉树遍历
输入两棵二叉树A, B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解析:
如何理解子结构?
可以理解成子结构是原树的子树,或者一部分。也就是说,B要是A的子结构,那么B的根结点+左子树+右子树,都在A中存在且构成树形结构。
如何判断B是否是A的子结构?
判断子结构的过程可以分为两步:
- 先确定起始位置。
- 判断从确定的位置开始,后续的左右子树的内容是否一致。
例如,对于以下两棵树,我们需要先在A树当中找到与B树根结点值相同的结点,然后再判断从该结点位置开始后续的左右子树的内容是否一致。
若判断结果一致,则直接得出B是A的子结构;若判断结果不一致,则重新在A树中寻找与B树根结点值相同的结点。如此进行下去,直到判定B是A的子结构或者A树被遍历完毕判定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 IsSame(TreeNode* begin, TreeNode* beginSub)
{
if (beginSub == nullptr) //beginSub为nullptr,说明已经比较完了
return true;
if (begin == nullptr) //beginSub未比较完,但begin已经比较完了
return false;
if (begin->val != beginSub->val) //结点值不相同,返回false
return false;
//结点值相同,且左子树和右子树相同则为相同的树
return IsSame(begin->left, beginSub->left) && IsSame(begin->right, beginSub->right);
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2) {
if (pRoot1 == nullptr || pRoot2 == nullptr) //pRoot1已经遍历完毕或pRoot2为空,则返回false
return false;
bool result = false;
//找到起始位置
if (pRoot1->val == pRoot2->val)
{
//从起始位置开始,判断两棵树是否相同
result = IsSame(pRoot1, pRoot2);
}
//从左子树寻找匹配的起始位置
if (result != true)
{
result = HasSubtree(pRoot1->left, pRoot2);
}
//从右子树寻找匹配的起始位置
if (result != true)
{
result = HasSubtree(pRoot1->right, pRoot2);
}
return result;
}
};