问题描述:
有两个不同大小的二进制树: T1
有上百万的节点; T2
有好几百的节点。请设计一种算法,判定 T2
是否为 T1
的子树。
注意事项
若 T1 中存在从节点 n 开始的子树与 T2 相同,我们称 T2 是 T1 的子树。也就是说,如果在 T1 节点 n 处将树砍断,砍断的部分将与 T2 完全相同。
实现思路:
判断是否为左右子树,首先要利用一个判断两颗树是否相等的函数。
如果T1为空则为假,否则如果T2为空则为真,否则如果T2和T1相等则为真,否则利用递归分别判断T2是否为T1左右子树的子树。
代码:
/**
* 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.
*/
bool isSubtree(TreeNode *T1, TreeNode *T2) {
// write your code here
if(!T2) return true;
else if(!T1) return false;
else if(IsEqual(T1,T2)) return true;
else if(isSubtree(T1->left,T2)) return true;
else return isSubtree(T1->right,T2);
}
bool IsEqual(TreeNode *T1,TreeNode *T2)
{ if(T1==NULL&&T2==NULL) {
return true;
} else if(T1==NULL^T2==NULL) {
return false;
}else if(T1->val!=T2->val){
return false;
} else {
return IsEqual (T1->left,T2->left)&&IsEqual(T1->right,T2->right);
}
}
};
感想:
判断是否为子树,如果是子树的话那么这个树肯定与T1的某个节点的子树完全相同,所以考虑到用一个判断两棵树是否相等的函数,考虑到用递归。