樹的子結構

 

題目：輸入兩棵二叉樹A，B，判斷B是不是A的子結構。（ps：我們約定空樹不是任意一個樹的子結構）

分析：先看B是不是以A的根結點爲根結點的子樹，是的話直接返回true,否則看是不是以A的左子樹爲根結點的子樹，再看右子樹這是一個遞歸的過程。

//判斷node2是不是以node1的根結點爲根結點的子樹
    bool is_subtree(TreeNode* node1, TreeNode* node2)
    {
        //第一次進來node2肯定不是空，因爲在主邏輯HasSubtree
        //中已經判斷過了
        
        if(node2==nullptr)//node2能夠走到空，說明node2之前的結點與node1已經匹配上了
            return true;
        
        if(node1==nullptr)//既然來到這，說明node2不爲空，那麼node1爲空，說明node2沒有匹配完
            return false;
        
        if(node1->val==node2->val)//如果兩個樹當前結點值相同,則要看兩個樹的左子樹與左子樹，右子樹與右子樹
            return is_subtree(node1->left, node2->left)
            && is_subtree(node1->right, node2->right);
        else
            return false;
    }
    
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        //只要有一棵樹是空樹，就不是子結構
        if(pRoot1==nullptr)
            return false;
        if(pRoot2==nullptr)
            return false;
            
        bool flag=is_subtree(pRoot1, pRoot2);//看pRoot2是不是以pRoot1根結點爲根結點的pRoot1的子樹
        if(flag==false)//說明不是，則看以pRoot1的左子樹爲根結點的是不是
            flag=is_subtree(pRoot1->left, pRoot2);
        if(flag==false)//說明不是，則看以pRoot1的右子樹爲根結點的是不是
            flag=is_subtree(pRoot1->right, pRoot2);
        
        return flag;
            
    }