题目:
输入一个整数数组,判断该数组是不是某二叉树的后序遍历的结果。如果是则返回true,否则返回false。假设输入的数组的任意两个数字都不相同。
思路:
这题看清楚了,是二叉搜索树,也就是带排序的。
仔细观察二叉搜索树的后序遍历,发现有下列特征:
1.以最后面的一个数字为参考值,则前面一部分小于这个值,后面一部分大于这个值。
2.或者前面的值全部大于或者小于这个值。
我们可以用递归来求解。
考虑数组:243675
则root肯定是5,左子树是243,右子树是67
对左子树,root肯定是3,左子树为2,右子树为3。
对右子树,root肯定是7,左子树为6,没有右子树。
我们可以先找到分界线,即开始大于最后一个数的位置pos。
定义函数bool isBinaryTreePos(vector<int> nums,int first,int last)进行递归。
#include <iostream>
#include <vector>
#include <queue>
#include <string>
#include <stack>
#include <algorithm>
using namespace std;
bool isSubBinaryTreePos(vector<int> nums, int first, int last)
{
if (first >= last || first + 1 == last) return true;
int pos = last - 1;
while (pos >= first && nums[pos] > nums[last]) pos--;
int index = pos;
while (index >= first && nums[index] < nums[last]) index--;
if (index >= first) return false;
return isSubBinaryTreePos(nums, first, pos) && isSubBinaryTreePos(nums, pos + 1, last - 1);
}
bool isBinaryTreePos(vector<int> nums)
{
int size = nums.size();
if (size < 3) return true;
return isSubBinaryTreePos(nums, 0, size - 1);
}
int main()
{
int arr[] = { 2, 4, 3, 6, 7, 5 };
vector<int> nums(arr, arr + 6);
cout << isBinaryTreePos(nums) << endl;
return 0;
}