子树

来源:互联网 发布:电影制作知乎 编辑:程序博客网 时间:2024/04/29 09:43

有两个不同大小的二叉树: 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

解题思路是:遍历T1寻找与T2根节点值相等的点。如果找到与T2根节点相同的点那就接着判断该点的左右子树的值是否与T2的左右子树的值相等。在此时需要注意的是,题目要求:如果在 T1 节点 n 处将树砍断,砍断的部分将与 T2 完全相同,也就是说T1不仅仅是包含T2,而且还要求在T2遍历到最底层时,T1也应该遍历到最底层,否则T2就不是T1的子树。

/**
 * 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: The roots of binary tree T1.
     * @param T2: The roots of binary tree T2.
     * @return: True if T2 is a subtree of T1, or false.
     */
     int flag1=0;
     int flag2=0;
    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(isSubtree2(T1,T2)){
                return true;
            }
        }
        if(T1->right!=NULL||T1->left!=NULL){
            return (isSubtree(T1->right,T2)||isSubtree(T1->left,T2));
        }
    }
    bool isSubtree2(TreeNode * T1, TreeNode * T2)
    {
        if(T1==NULL&&T2==NULL)
            return true;
        if(T1==NULL&&T2!=NULL)
            return false;
        if(T1!=NULL&&T2==NULL)//这句话的目的是:如果在 T1 节点 n 处将树砍断,砍断的部分将与 T2 完全相同。
            return false;
        if(T1!=NULL&&T2!=NULL&&(T1->val==T2->val)){
            if(isSubtree2(T1->left,T2->left)&&isSubtree2(T1->right,T2->right))
                return true;
            else return false;
        }
        else  return false;
    }
};