寻找树的子结构,C++
class Solution {
public:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
if (pRoot2 == nullptr)
return false;
queue<TreeNode*> tqu;
tqu.push(pRoot1);
while(!tqu.empty()){
int length = tqu.size();
for (int i = 0; i != length; ++i){
if (tqu.front() != nullptr){
tqu.push(tqu.front()->left);
tqu.push(tqu.front()->right);
if (pRoot2->val == tqu.front()->val){
// 执行判断子结构函数 若不是继续遍历pRoot1
if (IsSubtree(tqu.front(), pRoot2))
return true;
}
}
tqu.pop();
}
}
return false;
}
bool IsSubtree(TreeNode* pRoot1, TreeNode* pRoot2){
if (pRoot2 == nullptr)
return true;
else if (pRoot1 == nullptr || pRoot1->val != pRoot2->val)
return false;
else{
return IsSubtree(pRoot1->left, pRoot2->left) && IsSubtree(pRoot1->right, pRoot2->right);
}
}
};1、首先考虑题目提示,B为空树则不是子结构
2、接着层序遍历A,寻找A中与B头节点相同的节点
(1)找到后执行函数IsSubtree()递归函数,若判断B是子结构return true;判断不是则退出继续层序遍历
PS:题中没说A树中节点值均不相同,所以更深层可能有子结构
(2)层序完都没子结构则return false
欢迎大家交流指正!!!

