245.子树

来源:互联网 发布:vb automation 错误 编辑:程序博客网 时间:2024/06/05 07:51

题目:有两个不同大小的二进制树: T1 有上百万的节点; T2 有好几百的节点。请设计一种算法,判定 T2 是否为 T1的子树。


样例:

下面的例子中 T2 是 T1 的子树:

       1                3      / \              / T1 = 2   3      T2 =  4        /       4

下面的例子中 T2 不是 T1 的子树:

       1               3      / \               \T1 = 2   3       T2 =    4        /       4

代码:

class Solution {    public:    /**     * @param T1, T2: The roots of binary tree.     * @return: True if T2 is a subtree of T1, or false.     */    bool check(TreeNode *T1,TreeNode *T2){          if(!T1&&!T2)              return true;          if(!T1||!T2)              return false;          if(T1->val==T2->val){              return check(T1->left,T2->left)&&check(T1->right,T2->right);              }          return false;      }       bool isSubtree(TreeNode *T1, TreeNode *T2) {        // write your code here        if(!T2)              return true;          if(!T1)              return false;          if(T1->val==T2->val){              if(check(T1,T2))                  return true;          }          if(isSubtree(T1->left,T2))              return true;          if(isSubtree(T1->right,T2))              return true;          return false;      }};

感想:建立一个成员函数,用来检查两树或子树是否相等。在这个成员函数中,采用递归的思想,很容易做到。然后在判断第二个树是否为第一棵树的子树的函数中,除去两树可能为空的情况。然后调用检查函数,若两树相等,则为子树,否则,分别从第一棵树的左右子树开始分别递归调用检查函数,判断T1的左右子树中是否函数与T2相等的子树,利用递归的思想,很容易就能完成。

0 0
原创粉丝点击