问题描述
示例
思路
1、暴力
暴力解法就是针对原树中的每一个结点和其子树与标定的子树作对比,利用递归相当于回溯的概念
2、匹配过程中加入KMP
在匹配过程中,B数是A树的子串的话,说明符合条件。但是针对本题来说,可能会出现以下情况:
A根为1,左结点为2,右节点为NULL,B根为1,左结点为NULL,右节点为2,此时两者的先序遍历就是相同的,所以需要在原树中插入左空结点或者右空节点标识
(1)将A树的先序遍历以及B数的先序遍历存储起来
(2)利用KMP算法判断B是否是A的子串
代码(c++)
1、暴力
class Solution {
public:
bool check(TreeNode *o, TreeNode *t) {
if (!o && !t) return true;
if ((o && !t) || (!o && t) || (o->val != t->val)) return false;
return check(o->left, t->left) && check(o->right, t->right);
}
bool dfs(TreeNode *o, TreeNode *t) {
if (!o) return false;
//遍历每一个结点
return check(o, t) || dfs(o->left, t) || dfs(o->right, t);
}
bool isSubtree(TreeNode *s, TreeNode *t) {
return dfs(s, t);
}
};
2、加入匹配串
class Solution {
public:
vector <int> sOrder, tOrder;
int maxElement, lNull, rNull;
void getMaxElement(TreeNode *o) {
if (!o) return;
maxElement = max(maxElement, o->val);
getMaxElement(o->left);
getMaxElement(o->right);
}
void getDfsOrder(TreeNode *o, vector <int> &tar) {
if (!o) return;
tar.push_back(o->val);
if (o->left) getDfsOrder(o->left, tar);
else tar.push_back(lNull);
if (o->right) getDfsOrder(o->right, tar);
else tar.push_back(rNull);
}
bool kmp() {
int sLen = sOrder.size(), tLen = tOrder.size();
vector <int> fail(tOrder.size(), -1);
for (int i = 1, j = -1; i < tLen; ++i) {
while (j != -1 && tOrder[i] != tOrder[j + 1]) j = fail[j];//如果不符合条件,回溯到之前和torder[j]相同点的位置
if (tOrder[i] == tOrder[j + 1]) ++j;//判断当前点和j位置的点是否相等,如果相等,说明此时的j是当前串的一部分,t中0~j部分是s的子串
fail[i] = j;//无论是否是其子串,都要更新当前点为以后使用
}
//以上就更新了子串t的next数组(fail数组),即下标数组
for (int i = 0, j = -1; i < sLen; ++i) {//与以上同理
while (j != -1 && sOrder[i] != tOrder[j + 1]) j = fail[j];
if (sOrder[i] == tOrder[j + 1]) ++j;
if (j == tLen - 1) return true;//此步骤是为了设置限定条件,如果子串已经遍历结束,则退出
}
return false;
}
bool isSubtree(TreeNode* s, TreeNode* t) {
maxElement = INT_MIN;
getMaxElement(s);
getMaxElement(t);
lNull = maxElement + 1;
rNull = maxElement + 2;
getDfsOrder(s, sOrder);
getDfsOrder(t, tOrder);
return kmp();
}
};