題目描述:
輸入兩棵二叉樹A,B,判斷B是不是A的子結構。(ps:我們約定空樹不是任意一個樹的子結構)
解題思路:
同樣考慮用遞歸來做。
利用兩個遞歸函數,一個用于判斷兩棵樹樹否相等,另一個遞歸取A的子樹與B比較。
包含幾種情況:
A為空或B為空,false;
A與B相等,true;
B為A的子結構,true;
B不為A的子結構,false.
代碼:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
bool equal(TreeNode* t1, TreeNode* t2)
{
if(t1==nullptr && t2==nullptr)
return true;
else if(t1==nullptr)
return false;
else if(t2==nullptr)
return true;
else
{
if(t1->val==t2->val)
{
return (equal(t1->left, t2->left) && equal(t1->right, t2->right));
}
else
return false;
}
}
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if(pRoot2 == nullptr)
return false;
if(pRoot1 == nullptr)
return false;
if(pRoot1->val != pRoot2->val)
{
return (HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2));
}
else
{
if(equal(pRoot1->left, pRoot2->left)&& equal(pRoot1->right, pRoot2->right))
return true;
else
{
if(HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2))
return true;
else
return false;
}
}
}
};