题目:有两个不同大小的二进制树: T1
有上百万的节点; T2
有好几百的节点。请设计一种算法,判定 T2
是否为 T1
的子树。
样例:
下面的例子中 T2 是 T1 的子树:
1 3
/ \ /
T1 = 2 3 T2 = 4
/
4
下面的例子中 T2 不是 T1 的子树:
1 3
/ \ \
T1 = 2 3 T2 = 4
/
4
代码:
class Solution {
public:
/**
* @param T1, T2: The roots of binary tree.
* @return: True if T2 is a subtree of T1, or false.
*/
bool check(TreeNode *T1,TreeNode *T2){
if(!T1&&!T2)
return true;
if(!T1||!T2)
return false;
if(T1->val==T2->val){
return check(T1->left,T2->left)&&check(T1->right,T2->right);
}
return false;
}
bool isSubtree(TreeNode *T1, TreeNode *T2) {
// write your code here
if(!T2)
return true;
if(!T1)
return false;
if(T1->val==T2->val){
if(check(T1,T2))
return true;
}
if(isSubtree(T1->left,T2))
return true;
if(isSubtree(T1->right,T2))
return true;
return false;
}
};
感想:建立一个成员函数,用来检查两树或子树是否相等。在这个成员函数中,采用递归的思想,很容易做到。然后在判断第二个树是否为第一棵树的子树的函数中,除去两树可能为空的情况。然后调用检查函数,若两树相等,则为子树,否则,分别从第一棵树的左右子树开始分别递归调用检查函数,判断T1的左右子树中是否函数与T2相等的子树,利用递归的思想,很容易就能完成。