Lintcode 子树

来源:互联网 发布:tm域名查询 编辑:程序博客网 时间:2024/06/16 17:11

1:问题描述:

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

 注意事项

若 T1 中存在从节点 n 开始的子树与 T2 相同,我们称 T2 是 T1 的子树。也就是说,如果在 T1 节点 n 处将树砍断,砍断的部分将与 T2 完全相同。

2:解题思路:先写一个比较函数,用来比较两棵树是否相同,即是否结构相同且在相同位置的值一样,和判断两棵子树等价是一样的。用于判断子树的时候调用,判断T2是否为T1的子树时,就可以直接调用compare函数。只要T1中

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 compare(TreeNode *T1,TreeNode *T2)
    {
        if(T1==NULL&&T2==NULL)return 1;
        if(T1!=NULL&&T2==NULL||T1==NULL&&T2!=NULL)return 0;
        if(T1!=NULL&&T2!=NULL&&T1->val==T2->val)
        {
            return compare(T1->left,T2->left)&&compare(T1->right,T2->right);
        }
        return 0;
    }
    bool isSubtree(TreeNode *T1, TreeNode *T2)
    {
        // write your code here
        if(T1==NULL&&T2==NULL)return 1;
        if(T1==NULL)return 0;
        if(T2==NULL)return 1;
        if(T1->val==T2->val)
        {
          if(compare(T1,T2)) return 1;
        }
        if (isSubtree(T1->left,T2)) return 1;
        if(isSubtree(T1->right,T2)) return 1;
        return 0;
    }

};

4:解题感想:

  这个题比较难,自己想的时候一直不对,我一直想用一个函数来写但是一直不过,我一直觉得就是多分几类就可以,但是自己感觉自己写的就是不对的,也一直没过,在网上搜了个和我类似的答案,看了看别人写的,网上搜到的答案都是用了两个函数写的,先写一个判断函数,判断两个树是否相等,在判断子树的函数中调用就行,自己也懂了。


0 0
原创粉丝点击