1.描述
有两个不同大小的二进制树: 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
2.分析
第一步遍历T1找到节点值和T2根节点值相同的节点,第二步判断这两个节点的子树是否完全相同。
若相同即结束,若不同则继续在T1中寻找下一个和T2根节点值相同的节点,重复上述操作。
3.代码
/**
* 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 isequal(TreeNode *T1, TreeNode *T2)//该函数判断两个节点的子树是否相同,若相同返回true
{
if(T1==NULL&&T2==NULL)//若当前两个节点都为空,则当前两节点相同
return true;
if(T1!=NULL&&T2!=NULL&&T1->val==T2->val)//若当前两个节点相同,则判断它们相对应的孩子是否也相同
return isequal(T1->left,T2->left)&&isequal(T1->right,T2->right);
else
return false;
}
bool isSubtree(TreeNode *T1, TreeNode *T2){
// write your code here
if(T2==NULL)
return true;
if(T1==NULL)
return false;
bool equ;
equ=isequal(T1,T2);
if(!equ)//若当前两节点不同,T1的节点往后移继续和T2的根节点比较
equ=isSubtree(T1->left,T2);
if(!equ)
equ=isSubtree(T1->right,T2);
return equ;
}
};
4.总结
这里注意一点当从T1遍历找和T2的根节点相同的节点时只要T1的节点是在不断变化的,
找完当前节点找它的左右孩子,知道找到和T2的根节点相同的节点为止,在此过程中T2
是始终保持不变的。当从T1中找到一个节点和T2的根节点相同时,接下来的操作和
《LintCode 等价二叉树》的思想完全相同,判断两棵树是否结构及节点值相同。