子树-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




思路：

我们用判断两个树是否相等的函数，遍历T1的节点，判断每个节点的子树是否与T2相等。

这里说一下，我对有返回值的递归函数不怎么会用，下面 代码一 是一开始写的无返回值函数，用时120ms；

然后特别感谢下ww给我讲了有返回值的递归，代码二 用的是有返回值的递归~

代码一：

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

代码二：

/**
 * 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 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;
    }
    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;
        
    }
};