寻找树的子结构,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
欢迎大家交流指正!!!