子樹-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;
        
    }
};