题目取自LeetCode:
这道题让我们根据后序遍历结果来判断是否为合法的二叉搜索树。首先我们可以找一下遍历序列和二叉搜索树之间的关系,既然是后序遍历结果,那么我们可以知道数组中最后一个元素一定为二叉树的根节点,因此我们可以从这个方向入手编写程序。
在知道二叉搜索树的根节点后我们可以根据搜索树的性质(左子树一定小于根节点,右子树一定大于根节点)来判断是否为合法的二叉搜索树。
对于题目示例数组[1,6,3,2,5]来说其构成的二叉搜索树根节点必定为5,接下来找到其左右子树即可,顺序遍历数组可以得到左子树[1]和右子树[6,3,2],因为右子树中的3和2小于根节点5,因此返回false。
对于示例数组[1,3,2,6,5]来说,依次遍历可得左子树[1,3,2],右子树[6]符合二叉搜索树。再继续遍历左子树[1,3,2],其中2为左子树根节点,可得左子树[1]右子树[3]都符合题意,因此该数组返回true。
完整代码如下:
bool verifyPostorder(vector<int>& postorder) {
if(postorder.size()<=1)return true;//如果该树中只有一个结点或者无结点,则该子树为合法子树
int ii=postorder.size()-1;
for(int i=0;i<postorder.size()-1;i++){//找到左右子树分界点
if(postorder[i]>postorder[postorder.size()-1]){
ii=i;
break;
}
}
for(int i=ii+1;i<postorder.size()-1;i++){//判断右子树中元素是否全部大于根节点
if(postorder[i]<postorder[postorder.size()-1])return false;
}
vector<int>x1(postorder.begin(),postorder.begin()+ii);//拆分数组
vector<int>x2(postorder.begin()+ii,postorder.end()-1);
return verifyPostorder(x1)&&verifyPostorder(x2);
}