Lintcode---子树
有两个不同大小的二进制树: 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
思路:继续采用递归,借用等价二叉树判断的方法,先从根节点开始比较,若根节点相同,判断二叉树是否为等价二叉树;
若等价,则返回true,否则,若T1左子树或右子树不为空,对其左右子节点分别进行递归调用,判断是否存在等价二叉树。
之前没有借助等价二叉树判断,一直通过率为83%,应该是自身判断逻辑还有遗漏。
总结,确保逻辑正确,其他的交给计算机吧!
/** * 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. */ /* 思路:继续采用递归的方法,借用等价二叉树的方法,先从根节点开始比较,若根节点相同,判断二叉树是否为等价二叉树; 若等价,则返回true,否则,若T1左子树或右子树不为空,对其左右子节点分别进行递归调用,比较是否存在等价二叉树。 之前没有借助等价二叉树判断,一直通过率为83%,应该是自身判断逻辑还有遗漏。 总结,确保逻辑正确,其他的交给计算机吧! */ bool isSame(TreeNode* a, TreeNode* b) { // Write your code here if(a==NULL&&b==NULL){ return true; } if(a==NULL&&b!=NULL){ return false; } if(a!=NULL&&b==NULL){ return false; } if(a!=NULL&&b!=NULL&&(a->val==b->val)){ if(isSame(a->left,b->left)&&isSame(a->right,b->right)){ return true; } else{ return false; } } else{ return false; } } bool isSubtree(TreeNode *T1, TreeNode *T2) { // write your code here if(T1==NULL&&T2==NULL){ return true; } if(T1==NULL&&T2!=NULL){ return false; } if(T1!=NULL&&T2==NULL){ return true; } if(T1!=NULL&&T2!=NULL&&(T1->val==T2->val)){ if(isSame(T1,T2)){ return true; } } if(T1->right!=NULL||T1->left!=NULL){ return (isSubtree(T1->right,T2)||isSubtree(T1->left,T2)); } } };