JZ33 二叉搜索树的后序遍历序列

1. 问题描述

2. 样例说明

3.解法一：分治

算法思想

1. 二叉树的后序遍历顺序是：左子树 -> 右子树 -> 根节点，因此序列的最后一个数代表了根节点
2. 因此我们可以将一个序列划分为3段, 左子树+右子树+根, 例如[4, 8, 6, 12, 16, 14, 10]可以根据根节点的值将其划分为左子树[4, 8, 6], 右子树[12, 16, 14], 根[10], 由于我们是先确定的右子树区间, 因此当左子树区间中出现大于根节点的值时, 序列不合法。再采用分治的思想, 对于每段序列代表的子树, 检查它的左子树和右子树, 当且仅当左右子树都合法时返回true

代码

class Solution {
public:
    bool VerifySquenceOfBST(vector<int> sequence) {
       if(sequence.size()==0)  return false;  //空树则返回false
       int n=sequence.size();
       return check(sequence,0,n-1);
    }
    
    bool check(vector<int> sequence,int l,int r){
        if(l>=r) return true; //只有一个结点，或没有结点
        int root=sequence[r];  // 当前子树的根节点
        int j=r-1;
        while(j>=0&&sequence[j]>root) j--;  //划分右子树
        for(int i=l;i<=j;i++){  // 检查左子树是不是存在大于根节点的数
            if(sequence[i]>root) return false;
        }
        return check(sequence,l,j)&&check(sequence, j+1, r-1); // 分治法检查左子树和右子树
    }
};

时间空间复杂度分析

时间复杂度：O(n^2), n为二叉树节点的个数, 当树为链式时时间复杂度最坏为O(n^2)
空间复杂度：O(n), 当树为链式结构时, 递归深度为n。

4. 解法二：栈

算法思想

1. 实际上二叉树的中序遍历和后序遍历对应着一种栈的压入、弹出序列, 而对后序遍历序列从小到大排序就得到了中序遍历序列
2. 我们得到中序遍历序列后, 将其作为入栈序列, 检查后序遍历序列是不是一个合法的出栈序列即可

代码

class Solution {
public:
    bool VerifySquenceOfBST(vector<int> sequence) {
        if(sequence.empty()) return false;
        vector<int> inorder(sequence);
        sort(inorder.begin(), inorder.end());
        return IsPopOrder(inorder, sequence);
    }

    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
         int n = pushV.size();
         stack<int> stk;    // 使用STL中的栈容器
         int i = 0, j = 0;
         while(i < n){
             stk.push(pushV[i]);    // 首先将pushV[i]入栈
             while(!stk.empty() && stk.top() == popV[j]){    // 不断检查栈顶
                 ++j;
                 stk.pop();
             }
             ++i;
         }
         return j == n;    // 判断能否成功对应
     }
};

时间空间复杂度分析

时间复杂度：O(nlogn), 时间复杂度取决于排序
空间复杂度：O(n), 额外数组和辅助栈的空间