二叉搜索树的后序遍历序列.md

武培轩 2020年02月06日 88次浏览

题目描述

输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。

思路

对于后序遍历来说,序列数组的最后一个元素一定是根节点,则根据这个元素,将前面的数组分为左、右两个部分,左侧部分都小,右侧部分都大,如果右侧部分有比该根节点小的元素,那么就不是后序遍历,如此递归进行。

代码实现

package Tree;

/**
 * 二叉搜索树的后序遍历序列
 * 输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
 */
public class Solution53 {
    public static void main(String[] args) {
        Solution53 solution53 = new Solution53();
        int[] sequence = {5,7,6,9,11,10,8};
        System.out.println(solution53.VerifySquenceOfBST(sequence));
    }

    public boolean VerifySquenceOfBST(int[] sequence) {
        if (sequence == null || sequence.length == 0) return false;
        return verify(sequence, 0, sequence.length - 1);
    }

    private boolean verify(int[] sequence, int first, int last) {
        if (last - first <= 1) return true;
        int rootVal = sequence[last];
        int cutIndex = first;//cutIndex记录比rootVal(根节点)大的第一个元素的索引
        while (cutIndex < last && sequence[cutIndex] <= rootVal) cutIndex++;
        //如果右子树有元素小于root,返回false
        for (int i = cutIndex + 1; i < last; i++) {
            if (sequence[i] < rootVal) return false;
        }
        //递归求解左子树和右子树
        return verify(sequence, first, cutIndex - 1) && verify(sequence, cutIndex, last - 1);
    }
}