AcWing 46. 二叉搜索樹的後序遍歷序列

地址 https://www.acwing.com/solution/acwing/content/3959/

 

題目描述
輸入一個整數數組，判斷該數組是否是某二叉搜索樹的後序遍歷的結果。

若是是則返回true，不然返回false。

假設輸入的數組的任意兩個數字都互不相同。

樣例

輸入：[4, 8, 6, 12, 16, 14, 10]

輸出：true

根據二叉樹的定義來看 最後一個是根節點 前面的序列中必須存在一個拐點 拐點先後的值 要麼均大於根 要麼均小於根
那麼能夠將序列與根節點比較，遞歸逐步縮小序列範圍， 注意邊界條件 好比所有比根節點大或者所有比根節點小。
最後設置退出條件。

class Solution {
public:
    bool verifySequenceOfBSTInner(vector<int> sequence)
    {
        if(sequence.size() < 1) return true;
        int root =  sequence.back();  
        int flag = 0; int idx = -1;
        for(int i =0 ; i<sequence.size()-1;i++ ){
            if(sequence[i] > root && flag == 0){
                flag = 1;idx = i;
            }else if(sequence[i] < root && flag == 1){
                return false;
            }
        }

        bool ret1 = true;  vector<int> tmp1;
        if(idx > 0){
            tmp1 = vector<int>(sequence.begin(),sequence.begin()+idx);
            ret1  = verifySequenceOfBSTInner(tmp1); 
        }

        bool ret2 = true; vector<int> tmp2;

        if(idx < sequence.size()-1){
            tmp2 = vector<int>(sequence.begin()+idx,sequence.end()-1);
            ret2 = verifySequenceOfBSTInner(tmp2);
        }

        return ret1&& ret2;
    }

    bool verifySequenceOfBST(vector<int> sequence) {
        return verifySequenceOfBSTInner(sequence);
    }
};

做者：defddr
連接：https://www.acwing.com/solution/acwing/content/3959/
來源：AcWing
著做權歸做者全部。商業轉載請聯繫做者得到受權，非商業轉載請註明出處。