https://www.nowcoder.com/study/vod/1/3/2
对于两棵彼此独立的二叉树A和B,请编写一个高效算法,检查A中是否存在一棵子树与B树的拓扑结构完全相同。
给定两棵二叉树的头结点A和B,请返回一个bool值,代表A中是否存在一棵同构于B的子树。
1, 普通二叉树解法:遍历+匹配,B要么和A完全一样,要么是A的左子树的子树要么是A的右子树的子树。需要考察A中以每个节点开头的子树是否和B的头结点相同,如果相同则继续匹配孩子节点。如果不相同则继续考察A的左子树和右子树。时间复杂度是O(N*M),N是A的节点数M是B的节点数。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class IdenticalTree {
public:
bool isSame(TreeNode* A,TreeNode* B){
if(B == NULL && A == NULL)
return true;
if(A == NULL && B != NULL)
return false;
if(A != NULL && B == NULL)
return false;
if(A->val != B->val)
return false;
return isSame(A->left, B->left)&& isSame(A->right, B->right);
}
bool chkIdentical(TreeNode* A, TreeNode* B) {
// write code here
//只要B是空的就返回true
if(B==NULL) return true;
bool flag = false;
if(A)
if(A->val == B->val){
flag = isSame(A,B);
}
if(!flag) flag = chkIdentical(A->left,B);
if(!flag) flag = chkIdentical(A->right,B);
}
return flag;
}
};
2, 转化为字符串匹配问题,按照相同顺序遍历两个二叉树得到两个字符串,然后用字符串匹配算法KMP去判断第一个字符串中是否包含第二个字符串,如果包含则返回true否则false。在将数转化成字符串时,如果当前节点为空就用”#!”代替,访问完一个节点后添加一个”#”标志,以区分12和1 2这种情况。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class IdenticalTree {
public:
vector<int> get_next(string &t){
vector<int> next(t.length());
int i = -1;//前缀
int j = 0;//后缀
next[0] = -1;//初始化第一个字符的模式值
while(j < t.length()-1){//注意是t.length()-1,后缀表示当前字符的前一个
if(i==-1 || t[i] == t[j]){
i++;
j++;
next[j] = i;
}else{
//前缀回溯
i = next[i];
}
}
return next;
}
bool kmp(string &s,string &t){
int i=0,j=0;
vector<int> next = get_next(t);
while(s[i] && t[j]){
if(s[i] == t[j]){
i++;
j++;
}else if(j == 0){//j回溯到0位置,说明都不匹配,i+1继续下一轮
i++;
}else{
j = next[j];
}
}
if(!t[j]) return true;
return false;
}
void treeToString(TreeNode* t,string &str){
if(t==NULL){
str+="#!";
return;
}
str+=t->val;
str+="#";
treeToString(t->left,str);
treeToString(t->right,str);
}
bool chkIdentical(TreeNode* A, TreeNode* B) {
// write code here
string str_a;
string str_b;
treeToString(A,str_a);
treeToString(B,str_b);
return kmp(str_a,str_b);
}
};