题意:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
注意:在操作树的时候,我们一定要注意边界条件的检查,即检查空指针,当树A或树B为空的时候,定义相应的输出。
如果没有检查并进行相应的处理,则程序非常容易崩溃。
同时,还有一个细节值得注意:本题中节点中值的类型为double,在判断两个节点的值是不是相等的时候,不能直接使用==来进行判断,这是因为计算机在表示小数时(float,double)都有误差,判断两个小数是否相等,只能判断它们之差的绝对值是不是在一个很小的范围内,如果两个数相差很小,那么就可以认为它们相等。
//树的子结构,递归
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
const double eps = 0.0000001;
class Solution {
public:
Solution()
{
}
~Solution()
{
}
//小数比较大小
bool IsEqual(double a,double b)
{
if(fabs(a-b)<=eps) return true;
return false;
}
bool IsSubTree(TreeNode* pRoot1,TreeNode* pRoot2)
{
if(pRoot2==nullptr) return true; //到达叶节点
if(pRoot1==nullptr) return false;
if(pRoot1->val!=pRoot2->val) return false;
return IsSubTree(pRoot1->left,pRoot2->left)&&IsSubTree(pRoot1->right,pRoot2->right);
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(pRoot1==nullptr||pRoot2==nullptr) return false;
if(pRoot1!=nullptr&&pRoot2!=nullptr)
{
if(IsEqual(pRoot1->val,pRoot2->val)) {
//当前根节点值相同,比较子结构是否相同
if(IsSubTree(pRoot1,pRoot2)) return true;
}
if(HasSubtree(pRoot1->left,pRoot2)) return true;
if(HasSubtree(pRoot1->right,pRoot2)) return true;
}
return false;
}
};