题目
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。假设输入的数组的任意两个数字都互不相同。
思路
二叉搜索树后序遍历时,最后一个数字是树的根结点。数组中前面的数字分为两部分:第一部分是左子树结点的值,它们都比根结点的值小;第二部分是右子树结点的值,它们都比根结点大。
同理,左子树、右子树又可以看成一个小的二叉搜索树,用递归的方式可以判断,左子树结点的值是否小于根结点,右子树结点的值是否大于根结点。同时满足,结点为真。
代码
#include <iostream>
#include <queue>
using namespace std;
bool VerifySequenceOfBST(int sequence[], int length)
{
if (sequence == nullptr || length <= 0)
return false;
int root = sequence[length-1];
int i = 0;
for (; i < length - 1; ++i)
{
if (sequence[i] > root)
break;
}
int j = i;
for (; j < length - 1; ++j)
{
if (sequence[j] < root)
return false;
}
bool left = true;
if (i > 0)
left = VerifySequenceOfBST(sequence, i);
bool right = true;
if (i < length - 1)
right = VerifySequenceOfBST(sequence + i, length - i - 1);
return (left && right);
}
int main()
{
int binary_sort_tree[] = {5,7,6,9,11,10,8};
cout << "The result is "
<< VerifySequenceOfBST(binary_sort_tree, sizeof(binary_sort_tree)/ sizeof(int)) << endl;
return EXIT_SUCCESS;
}