1119. Pre- and Post-order Traversals (30)

题目链接：https://www.patest.cn/contests/pat-a-practise/1119

以题目给的例子说以下思路：
一、[1, 2, 3, 4, 6, 7, 5] [2, 6, 7, 4, 5, 3, 1]
1、前序遍历的第一个，后序遍历的最后一个节点的元素都为1，这是根节点
2、接下来判断该树是否唯一。
方法： 前序遍历完根节点后，应该会去遍历左子树；后序遍历最后一个元素是根节点，那么之前的元素应该是右子树的。这是左子树和右子树都存在的情况下。那么前序遍历的第二个节点应该是左子树的根节点，后序遍历的倒数第二个节点应该是右子树的根节点。如果这两个元素相同的话，这棵树就不唯一；不相同的话需要继续递归从而进行判断。
前序遍历的第二个节点为2，后序遍历的倒数第二个节点是3，所以目前为止这棵树是唯一的。
3、经过步骤2找到了根节点的左，右子树的”根节点”。之后需要确定左子树的前序遍历的序列，左子树的后序遍历的序列，然后进行递归建树的过程。
右子树的根节点元素为3，左子树的根节点元素为2。左子树的前序遍历序列为序列[1,2,3,4,6,7,5]中2到3（不包括3）之间的元素，即[2]，左子树的后序遍历的为[2,6,7,4,5,3,1]为元素2(包括)之前的元素序列，即[2]
4、根据左子树的前，后遍历的序列，就可以知道右子树的前、后遍历的序列然后递归建树。
在本例中，右子树的前序遍历的序列为[1,2,3,4,6,7,5]中3（包括）之后的序列，即[3,4,6,7,5]；右子树的后序遍历序列为[2,6,7,4,5,3,1] 中2（不包括）到3之间的元素，即[6,7,4,5,3]
二、在列二中
1 2 3 4
2 4 3 1
根节点为1，左，右子树的节点元素为2，3。但是在右子树的前序遍历序列[3,4]第二个元素和后序遍历序列[4,3]中倒数第二个元素相同，所以这棵树不唯一。

实现代码：

#include <iostream>#include <malloc.h>#include <vector>using namespace std;#define N 33typedef struct BTree{    struct BTree* left;    int data;    struct BTree* right;}BTree;bool isUnique=true;//用来存储该树是否唯一vector<int> inseq;//存储中序遍历的序列BTree* buildTree(int* pre,int* post,int n)//pre,post分别为一棵树的前序，后序遍历{//那么前序第一个，后序的最后一个一定相等，因为该节点为根节点    if(n==0)//递归结束的条件        return NULL;    BTree* root=(BTree*)malloc(sizeof(BTree));    root->data=pre[0];//前序遍历的第一个节点（后序的最后一个节点）为根节点    if(1<n && n-2>=0 && pre[1]==post[n-2])//前序的第二个，后序的倒数第二个相等        isUnique=false;//则该树不唯一    int leftEnd=1;//记录左子树前序遍历的终止位置。下标1为开始的位置    while(leftEnd<n && pre[leftEnd]!=post[n-2])        leftEnd++;    //递归左子树,需要找到左子树的前序，后序遍历序列    root->left=buildTree(pre+1,post,leftEnd-1);    //递归右子树,右子树的前序，后序序列    root->right=buildTree(pre+leftEnd,post+leftEnd-1,n-leftEnd);    return root;}void inOrder(BTree* root)//中序遍历{    if(root)    {        inOrder(root->left);        inseq.push_back(root->data);        inOrder(root->right);    }}int main(){    int n;    int pre[N],post[N];    cin>>n;    for(int i=0;i<n;i++)        cin>>pre[i];    for(int i=0;i<n;i++)        cin>>post[i];    BTree* root=buildTree(pre,post,n);//建树    inOrder(root);//中序遍历    if(isUnique)        cout<<"Yes"<<endl;    else        cout<<"No"<<endl;    for(int i=0;i<inseq.size();i++)    {        if(i==inseq.size()-1)            cout<<inseq[i]<<endl;        else            cout<<inseq[i]<<" ";    }}