题目描述
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
解题思路
首先默认大家知道什么是二叉搜索树的后序遍历,大概就是遍历一个树的时候,先遍历左节点,然后遍历右节点,最后输出根节点,不明白的可以百度一下查一下相关的例子就能够理解了,应该是还比较好理解的。
那么怎么判断一个序列是不是二叉搜索树的后序遍历序列呢?
首先要考虑到二叉搜索树的性质,就是任意节点的左子树的元素都小于这个节点,右子树的元素都大于这个节点。那么我们可以从这个性质着手来解决问题。
一个序列,最后一个数字一定是作为根节点,然后前一部分一定为左子树,后一部分一定为右子树,也就是说, 2 1 4 5 3这个序列在这个节点是可以的,这个节点为3,左子树为2 1,右子树为4 5,而 3 1 2则是不可以的,因为3是大于2的,所以3 1按道理都是2的右子树,这个时候1是小于2的,不满足二叉搜索树的性质,所以在这个节点上不满足二叉搜索树的性质。
知道上述的思路之后,后面就简单了,就是不断的递归,最后返回结果,首先检查根节点是不是满足性质,然后检查根节点的左右子树是否满足性质,一直检查到最后检查了整个树的每一个节点,如果都满足性质则返回true,否则返回false。
具体代码如下:
class Solution
{
public:
bool VerifySquenceOfBST(vector<int> sequence)
{
//LRD 先左后右、再根节点
if (sequence.size() == 0) return false;
if (sequence.size() == 1) return true;
bool flag = false;
int tmp = -1;
vector<int> left{};
vector<int> right{};
for (int i = 0; i < sequence.size() - 1; i++)
{
if (sequence[i] > sequence.back())
{
flag = true;
tmp = i;
}
if (flag && sequence[i] < sequence.back()) return false;
if (flag == false) left.push_back(sequence[i]);
if (flag == true) right.push_back(sequence[i]);
}
if (left.empty()) return VerifySquenceOfBST(right);
if (right.empty()) return VerifySquenceOfBST(left);
return VerifySquenceOfBST(right) && VerifySquenceOfBST(left);
}
};