有两个不同大小的二进制树: T1
有上百万的节点; T2
有好几百的节点。请设计一种算法,判定T2
是否为 T1
的子树。
注意事项
若 T1 中存在从节点 n 开始的子树与 T2 相同,我们称 T2 是 T1 的子树。也就是说,如果在 T1 节点 n 处将树砍断,砍断的部分将与 T2 完全相同。
您在真实的面试中是否遇到过这个题?
Yes
样例
下面的例子中 T2 是 T1 的子树:
1 3
/ \ /
T1 = 2 3 T2 = 4
/
4
下面的例子中 T2 不是 T1 的子树:
1 3
/ \ \
T1 = 2 3 T2 = 4
/
4
思路:继续采用递归,借用等价二叉树判断的方法,先从根节点开始比较,若根节点相同,判断二叉树是否为等价二叉树;
若等价,则返回true,否则,若T1左子树或右子树不为空,对其左右子节点分别进行递归调用,判断是否存在等价二叉树。
之前没有借助等价二叉树判断,一直通过率为83%,应该是自身判断逻辑还有遗漏。
总结,确保逻辑正确,其他的交给计算机吧!
/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
*/
class Solution {
public:
/**
* @param T1, T2: The roots of binary tree.
* @return: True if T2 is a subtree of T1, or false.
*/
/*
思路:继续采用递归的方法,借用等价二叉树的方法,先从根节点开始比较,若根节点相同,判断二叉树是否为等价二叉树;
若等价,则返回true,否则,若T1左子树或右子树不为空,对其左右子节点分别进行递归调用,比较是否存在等价二叉树。
之前没有借助等价二叉树判断,一直通过率为83%,应该是自身判断逻辑还有遗漏。
总结,确保逻辑正确,其他的交给计算机吧!
*/
bool isSame(TreeNode* a, TreeNode* b) {
// Write your code here
if(a==NULL&&b==NULL){
return true;
}
if(a==NULL&&b!=NULL){
return false;
}
if(a!=NULL&&b==NULL){
return false;
}
if(a!=NULL&&b!=NULL&&(a->val==b->val)){
if(isSame(a->left,b->left)&&isSame(a->right,b->right)){
return true;
}
else{
return false;
}
}
else{
return false;
}
}
bool isSubtree(TreeNode *T1, TreeNode *T2) {
// write your code here
if(T1==NULL&&T2==NULL){
return true;
}
if(T1==NULL&&T2!=NULL){
return false;
}
if(T1!=NULL&&T2==NULL){
return true;
}
if(T1!=NULL&&T2!=NULL&&(T1->val==T2->val)){
if(isSame(T1,T2)){
return true;
}
}
if(T1->right!=NULL||T1->left!=NULL){
return (isSubtree(T1->right,T2)||isSubtree(T1->left,T2));
}
}
};