有两个不同大小的二进制树: T1 有上百万的节点; T2 有好几百的节点。请设计一种算法,判定 T2 是否为 T1的子树。
注意事项
若 T1 中存在从节点 n 开始的子树与 T2 相同,我们称 T2 是 T1 的子树。也就是说,如果在 T1 节点 n 处将树砍断,砍断的部分将与 T2 完全相同。
您在真实的面试中是否遇到过这个题? Yes
这个题其实不好思考,其中嵌套两层递归,需要清楚程序的运行。
解题思路:这种题第一选择肯定是递归处理,节点值不同时,T1向下走,节点相等时,开始判断是否两个树结构相同,不同则返回错误,相同则返回正确。想起来简单,但是代码却不容易写出(对鄙人来说,不乏有强人存在)。下面我用代码进行详解。
bool isSubtree(TreeNode *T1, TreeNode *T2)
{
bool ret = false;
if(T2 == NULL)
return true;
if(T1 == NULL)
return false;
if(T1->val == T2->val)
{
ret = check(T1,T2); //check判断T1,T2结构是否相同
}
if(!ret) //如果之前结构不同,T1继续递归
ret = isSubtree(T1->left,T2);
if(!ret)
ret = isSubtree(T1->right,T2);
return ret;
}
bool check(TreeNode* T1,TreeNode* T2)
{
if(!T1 && !T2)
return NULL;
if(!T1 || !T2)
return false;
if(T1->val == T2->val) //节点相等,T1,T2继续遍历
return check(T1->left,T2->left)&&check(T1->right,T2->right);
return false;
}