题目描述:子树
问题描述:有两个不同大小的二进制树: T1 有上百万的节点; T2 有好几百的节点。请设计一种算法,判定 T2 是否为 T1的子树。注意事项:若 T1 中存在从节点 n 开始的子树与 T2 相同,我们称 T2 是 T1 的子树。也就是说,如果在 T1 节点 n 处将树砍断,砍断的部分将与 T2 完全相同。
思路:判断一棵树T1是否为另一棵树T2的子树,首先在T1中判断找到T2的根对应的节点,然后再判断这个根的左右子树是否与T2相同。
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 flag=0;
if(T2==NULL)
return 1;
if(T1==NULL)
return 0;
//if(T1->val==T2->val)
flag=dfs(T1,T2);
if(!flag)
flag=isSubtree(T1->left,T2);
if(!flag)
flag=isSubtree(T1->right,T2);
return flag;
}
bool dfs(TreeNode* a, TreeNode* b)
{
if(a==NULL && b==NULL)
return 1;
if(a!=NULL && b!=NULL && a->val==b->val)
{
return dfs(a->left,b->left) && dfs(a->right,b->right);
}
return 0;
}
};