面试题26:树的子结构
题意:输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
不难吧,只要递归的姿势正确,头指针注意一下,问题不大。
class Solution {
public:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool result = false;
if(pRoot1 != NULL && pRoot2 != NULL)
{
if(pRoot1->val == pRoot2->val)
result = doesTree1HaveTree2(pRoot1,pRoot2);
if(!result)
result = HasSubtree(pRoot1->left,pRoot2);
if(!result)
result = HasSubtree(pRoot1->right,pRoot2);
}
return result;
}
bool doesTree1HaveTree2(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(pRoot2 == NULL) return true;
if(pRoot1 == NULL) return false;
if(pRoot1 -> val != pRoot2 -> val) return false;
return doesTree1HaveTree2(pRoot1->left,pRoot2->left)&&doesTree1HaveTree2(pRoot1->right,pRoot2->right);
}
};