题目:输入两棵二叉树A和B,判断B是不是A的子结构。二叉树的节点的定义如下:
struct BinaryTreeNode
{
double m_dbValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
};
思路:
这道题涉及到树的遍历,首先我们要在A树中找到与B树根节点值相同的节点,接着对每一个找到的节点进行判断,看是否有和B树中相同的结构。由于里面的指针比较多,要随时注意输入的指针为空的情况,避免程序发生崩溃。其中递归的方法要好好理解。还有浮点数的比较不能直接用等号,而是要判断它们绝对值的差是不是在一个很小的范围内。
代码:
struct BinaryTreeNode
{
double m_dbValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
};
bool Equal(double num1, double num2)
{
if ((num1 - num2 > -0.0000001) && (num1 - num2 < 0.0000001))
return true;
else
return false;
}
bool DoesTreeHaveTree2(BinaryTreeNode* pRoot1, BinaryTreeNode* pRoot2)
{
if (pRoot2 == nullptr)
return true;
if (pRoot1 == nullptr)
return false;
if (!Equal(pRoot1->m_dbValue, pRoot2->m_dbValue))
return false;
return DoesTreeHaveTree2(pRoot1->m_pLeft, pRoot2->m_pLeft) && DoesTreeHaveTree2(pRoot1->m_pRight, pRoot2->m_pRight);
}
bool HasSubtree(BinaryTreeNode* pRoot1, BinaryTreeNode* pRoot2)
{
bool result = false;
if (pRoot1 != nullptr&&pRoot2 != nullptr)
{
if(Equal(pRoot1->m_dbValue,pRoot2->m_dbValue))
result=DoesTreeHaveTree2(pRoot1,pRoot2);
if (!result)
result = HasSubtree(pRoot1->m_pLeft, pRoot2);
if (!result)
result = HasSubtree(pRoot2->m_pRight, pRoot2);
}
return result;
}
复习:
这道题用了两个递归函数来实现,思路很清晰,还要多做几遍。
二刷代码:
struct BinaryTreenode
{
double m_dbValue;
BinaryTreenode* m_pLeft;
BinaryTreenode* m_pRight;
};
bool equal(double num1, double num2)
{
if (num1 - num2<0.0000001&&num1 - num2>-0.0000001)
return true;
else
return false;
}
bool DoseTree1HasTree2(BinaryTreenode* pRoot1, BinaryTreenode* pRoot2)
{
if (pRoot2 == nullptr)
return true;
if (pRoot1 == nullptr)
return false;
if (!equal(pRoot1->m_dbValue, pRoot2->m_dbValue))
return false;
return DoseTree1HasTree2(pRoot1->m_pLeft, pRoot2->m_pLeft) && DoseTree1HasTree2(pRoot1->m_pRight, pRoot2->m_pRight);
}
bool hasSubtree(BinaryTreenode* pRoot1, BinaryTreenode* pRoot2)
{
bool result = false;
if (pRoot1 != nullptr&&pRoot2 != nullptr)
{
if (equal(pRoot1->m_dbValue, pRoot2->m_dbValue))
result = DoseTree1HasTree2(pRoot1, pRoot2);
if (!result)
result = DoseTree1HasTree2(pRoot1->m_pLeft, pRoot2);
if (!result)
result = DoseTree1HasTree2(pRoot1->m_pRight, pRoot2);
}
return result;
}