1.题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
2.解题思路
首先要进行程序的鲁棒性分析:
1. 如果B为空树,那么判断结果为false,因为空树不是任意一棵树的子结构;
2. 如果A为空树,那么判断结果也为false。
如此分析以后,只需对A与B均不是空树的情况进行分析。分析思路如下:
1)判断当前A节点和B树的根节点是否相等
2)如果当前节点相等,那么继续对二者当前节点的左右子树进行判断。如果左右子树不等,那么就不是子结构。
3)如果步骤1)中二者当前节点不等,那么则在A中选择左子树作为新的根节点进行步骤1)。
4)如果在左子树中未能判定为子结构,那么则挑选A的右孩子作为新的根节点进行判断。
3.解答——cpp
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
// help
bool IsSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if (pRoot2 == nullptr){
return true;
}
if (pRoot1 == nullptr && pRoot2 != nullptr){
return false;
}
if (pRoot1->val != pRoot2->val){
return false;
}
return (IsSubtree(pRoot1->left,pRoot2->left) && IsSubtree(pRoot1->right,pRoot2->right));
}
// Original and Robuness
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool result = false;
// 空树不是任意一个树的子结构,即B为空树时直接返回false
if (pRoot2 == nullptr)
return false;
// 当A为空树且B非空时,B不是A的子结构
if (pRoot1 == nullptr && pRoot2 != nullptr)
return false;
// 当A、B都不是空树时
// 调用IsSubtree判定B是不是A的子结构
// 如果当前指向的节点相等,则调用Is继续判断
if (pRoot1->val == pRoot2->val){
result = IsSubtree(pRoot1,pRoot2);
}
// 如果当前节点判定失败,则将当前节点的左子树作为新的根节点重新判断
if (!result){
result = HasSubtree(pRoot1->left,pRoot2);
}
// 如果左子树判定也失败,则选择右子树判定
if (!result){
result = HasSubtree(pRoot1->right,pRoot2);
}
// 返回最终结果
return result;
}
};