问题描述:
有两个不同大小的二进制树: T1
有上百万的节点; T2
有好几百的节点。请设计一种算法,判定 T2
是否为 T1
的子树。
注意事项
若 T1 中存在从节点 n 开始的子树与 T2 相同,我们称 T2 是 T1 的子树。也就是说,如果在 T1 节点 n 处将树砍断,砍断的部分将与 T2 完全相同
下面的例子中 T2 是 T1 的子树:
1 3
/ \ /
T1 = 2 3 T2 = 4
/
4
下面的例子中 T2 不是 T1 的子树:
1 3
/ \ \
T1 = 2 3 T2 = 4
/
4
解题思路:
定义一个bool类型的变量,用以判断T2是否为T1的子树。分情况进行讨论,若T2为空,则T2定为T1的子树;若T1为空,则T2定不是T1的子树。如果T1和T2都不为空时,先找到T1中与T2相同的节点,如果T1的根节点不等于T2的根节点时,在T1的左子树和右子树中分别查找与T2的根节点相同的节点,当找到与T2的根节点相同的节点后再用递归继续判断对应的左子树和右子树数值是否相同,只有对应的左子树和右子树都相同时,才可以断定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
bool z=false;
if(T2==NULL) return true;
if(T1==NULL&&T2!=NULL) return false;
if(T1->val==T2->val){
z=func(T1,T2);
}
if(z==false){
z=isSubtree(T1->left,T2);
}
if(z==false){
z=isSubtree(T1->right,T2);
}
return z;
}
bool func(TreeNode *T1,TreeNode *T2){
if(T1!=NULL&&T2!=NULL&&T1->val==T2->val){
return func(T1->left,T2->left)&&func(T1->right,T2->right);
}
if(T1==NULL&&T2==NULL){
return true;
}
return false;
}
};
解题感悟:
解这个题目需要很强的逻辑,过程很复杂,尤其要注意找到根节点相同的位置才能向下继续判断。