【问题】
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
【分析】
这个问题其实不难,不过要注意代码上递归的技巧,并不是所有情况都是和我们用一个递归函数来解决的,这道题就用分类递归的思想,很让我受益。
【代码如下】
class Solution {
public:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(pRoot1 == NULL || pRoot2 == NULL)
return false;
else
return is_subtree(pRoot1, pRoot2) //is_subtree vs has_subtree
|| HasSubtree(pRoot1->left, pRoot2)
|| HasSubtree(pRoot1->right, pRoot2);
}
private:
bool is_subtree(TreeNode* t1, TreeNode* t2){
if(t2 == NULL)
return true;
else if(t1 == NULL)
return false;
if(t1->val == t2->val){
return is_subtree(t1->left, t2->left)
&& is_subtree(t1->right, t2->right);
}
else
return false;
}
};
将 HasSubtree 转化为结点值相等时的递归方法 is_subtree(),以及结点值不等时,在原函数 HasSubtree() 自身上递归。这种分类递归,需要多注意。